Rate this Page
★★★★★
torch.is_complex#
- torch.is_complex(input:Tensor)→bool#
Returns True if the data type of
inputis a complex data type i.e.,one oftorch.complex64, andtorch.complex128.- Parameters
input (Tensor) – the input tensor.
Example:
>>>torch.is_complex(torch.tensor([1,2,3],dtype=torch.complex64))True>>>torch.is_complex(torch.tensor([1,2,3],dtype=torch.complex128))True>>>torch.is_complex(torch.tensor([1,2,3],dtype=torch.int32))False>>>torch.is_complex(torch.tensor([1.0,2.0,3.0],dtype=torch.float16))False
On this page
