torch.is_complex

torch.is_complex(input)

Возвращает True, если тип данных input — комплексный тип данных, то есть один из torch.complex64, и torch.complex128.

Параметры:

input (Тензор) — входной тензор.