torch.is_complex

torch.is_complex(input)

如果 input 的資料型別是複數資料型別，即 torch.complex64 或 torch.complex128，則返回 True。

引數

input (張量) – 輸入的張量。