Method test_tensor_cuda_input

tests/transforms/test_to_numpy.py:58–65

(self)

    @skip_if_no_cuda
    def test_tensor_cuda_input(self):
        test_data = torch.tensor([[1, 2], [3, 4]]).cuda()
        test_data = test_data.rot90()
        self.assertFalse(test_data.is_contiguous())
        result = ToNumpy()(test_data)
        self.assertIsInstance(result, np.ndarray)
        self.assertTrue(result.flags["C_CONTIGUOUS"])
        assert_allclose(result, test_data, type_test=False)

    def test_list_tuple(self):
        test_data = [[1, 2], [3, 4]]