torch.is_complex
- torch.is_complex(input)
Returns True if the data type of
input
is a complex data type i.e., one oftorch.complex64
, andtorch.complex128
.- Parameters:
input (Tensor) – the input tensor.
Returns True if the data type of input
is a complex data type i.e.,
one of torch.complex64
, and torch.complex128
.
input (Tensor) – the input tensor.