mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 21:14:14 +08:00
02ab6ced8e
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/34697 Differential Revision: D20596860 Pulled By: anjali411 fbshipit-source-id: 18599fce5bd3513be17ecf83ba9fb0d64d971fc4
14 lines
489 B
Python
14 lines
489 B
Python
import torch
|
|
from torch.testing._internal.common_utils import TestCase, run_tests
|
|
|
|
devices = (torch.device('cpu'), torch.device('cuda:0'))
|
|
|
|
class TestComplexTensor(TestCase):
|
|
def test_to_list_with_complex_64(self):
|
|
# test that the complex float tensor has expected values and
|
|
# there's no garbage value in the resultant list
|
|
self.assertEqual(torch.zeros((2, 2), dtype=torch.complex64).tolist(), [[0j, 0j], [0j, 0j]])
|
|
|
|
if __name__ == '__main__':
|
|
run_tests()
|