torch.is_complex

torch.is_complex(input)

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

Parameters

input (Tensor) – the input tensor.