(self)
| 801 | assert ret_indices is indices |
| 802 | |
| 803 | def test_cat_out(self): |
| 804 | a = torch.tensor([1, 2], device=device) |
| 805 | b = torch.tensor([3, 4], device=device) |
| 806 | out = torch.empty(4, device=device) |
| 807 | ret = torch.cat([a, b], out=out) |
| 808 | np.testing.assert_equal(out.cpu().numpy(), [1, 2, 3, 4]) |
| 809 | assert ret is out |
| 810 | |
| 811 | def test_cat_out_empty_1d(self): |
| 812 | # Test tiny and cpu to show test passes on torch cpu |
nothing calls this directly
no test coverage detected