Skip to content

Commit

Permalink
shapesizes
Browse files Browse the repository at this point in the history
  • Loading branch information
rcoreilly committed Jan 8, 2025
1 parent 4d2aabf commit 92bdc5a
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion axon/layer_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -33,7 +33,7 @@ func TestLayer(t *testing.T) {
// can't have spiked as we haven't run the network yet
assert.Equal(t, float32(0.0), tensor.Values[i])
}
assert.Equal(t, []int{2, 2}, tensor.Shape().Sizes)
assert.Equal(t, []int{2, 2}, tensor.ShapeSizes())
}

/*
Expand Down

0 comments on commit 92bdc5a

Please sign in to comment.