get_tensor(x) will return a tensor with shape (x,) not (1, x), so there is a shape mismatch.
Not sure which is meant to be the correct shape, but I'd assume the shape should be (out_features,) to match the tests.
Although, this line seemed to have been edited recently, so I'm not sure if I've got this wrong.
The notebook describes the bias shape as
(1, out_features)
However, in the tests, the bias is manually set as so:
get_tensor(x)
will return a tensor with shape(x,)
not(1, x)
, so there is a shape mismatch.Not sure which is meant to be the correct shape, but I'd assume the shape should be
(out_features,)
to match the tests. Although, this line seemed to have been edited recently, so I'm not sure if I've got this wrong.