torch.is_complex()

torch.is_complex(input: Tensor) -> bool

Return True if the data type of input is a complex data type i.e., one of torch.complex64, and torch.complex128.

 

import torch

t = torch.tensor([1,2,3], dtype=torch.complex64) # True
print(torch.is_complex(t))
t = torch.tensor([1,2,3], dtype=torch.int32)     # False 
print(torch.is_complex(t))

 

❓torch.complex64, torch.complex128?

PyTorch에서 복소수(complex number)를 다루기 위한 dtype이 torch.complex64, torch.complex128이다.