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.

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