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.