torch.ones(2,2)*2
tensor([[2., 2.],
[2., 2.]])
torch.ones(2,2)*2j
tensor([[0., 0.],
[0., 0.]])
torch.tensor([[1,1],[1,1]], dtype=torch.complex64)2j
tensor([[(0.0000 + 2.0000j), (0.0000 + 2.0000j)],
[(0.0000 + 2.0000j), (0.0000 + 2.0000j)]], dtype=torch.complex64)
torch.ones(2,2)(1+2j)
tensor([[1., 1.],
[1., 1.]])
cc @ezyang @anjali411 @dylanbespalko