MCPcopy
hub / github.com/nlpodyssey/spago / WithShape

Function WithShape

mat/dense_new.go:28–32  ·  view source on GitHub ↗
(shape ...int)

Source from the content-addressed store, hash-verified

26}
27
28func WithShape(shape ...int) OptionsFunc {
29 return func(opts *Options) {
30 opts.Shape = shape
31 }
32}
33
34func WithBacking[T float.DType](data []T) OptionsFunc {
35 return func(opts *Options) {

Callers 15

testReshapeForwardFunction · 0.92
testAddScalarForward2Function · 0.92
BackwardMethod · 0.92
testStackForwardFunction · 0.92
BackwardMethod · 0.92
testFlattenForwardFunction · 0.92
testDotForwardFunction · 0.92
testMaxPoolForwardFunction · 0.92
testRowForwardFunction · 0.92
BackwardMethod · 0.92
testAffineFunction · 0.92
testAtForwardFunction · 0.92

Calls

no outgoing calls

Tested by 15

testReshapeForwardFunction · 0.74
testAddScalarForward2Function · 0.74
testStackForwardFunction · 0.74
testFlattenForwardFunction · 0.74
testDotForwardFunction · 0.74
testMaxPoolForwardFunction · 0.74
testRowForwardFunction · 0.74
testAffineFunction · 0.74
testAtForwardFunction · 0.74
testSliceForwardFunction · 0.74
testMulForwardMatrixMatrixFunction · 0.74
testMulForwardMatrixVectorFunction · 0.74