(self)
| 573 | ) |
| 574 | |
| 575 | def test_transpose_noargs(self): |
| 576 | x = mx.array([[0, 1, 1], [1, 0, 0]]) |
| 577 | |
| 578 | expected = [ |
| 579 | [0, 1], |
| 580 | [1, 0], |
| 581 | [1, 0], |
| 582 | ] |
| 583 | |
| 584 | self.assertListEqual(mx.transpose(x).tolist(), expected) |
| 585 | |
| 586 | def test_transpose_axis(self): |
| 587 | x = mx.array( |