()
| 277 | |
| 278 | |
| 279 | def test_transpose(): |
| 280 | x = np.arange(10).reshape(2, 5).astype("int32") |
| 281 | xx = Tensor(x) |
| 282 | (yy,) = transpose(xx, pattern=[1, -1, 0]) |
| 283 | np.testing.assert_equal(np.expand_dims(x.transpose(), axis=1), yy.numpy()) |
| 284 | |
| 285 | |
| 286 | def test_broadcast(): |