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.

© 2024, PyTorch Contributors
PyTorch has a BSD-style license, as found in the LICENSE file.
https://pytorch.org/docs/1.13/generated/torch.is_complex.html