torch.is_complex

torch.is_complex(input)

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

Параметры

input (Tensor) – входной тензор.