search

dlsyscourse / hw1

7 stars 27 forks source link

I think there is a problem at line 205 to line 209 of test_autograd_hw.py #13

Closed 0324wy closed 1 year ago

0324wy commented 1 year ago 
np.testing.assert_allclose(ndl.transpose(ndl.Tensor([[[2.45]],
   [[3.5 ]],
   [[0.9 ]]])).numpy(), np.array([[[2.45],
    [3.5 ],
    [0.9 ]]]))

Aims to transpose (3, 1, 1) => (1, 3, 1), but not give axes = (0, 1). The default axes are the last two axes(1, 2).