is_complex

paddle. is_complex ( x ) [源代码]

判断输入 tensor 的数据类型是否为复数类型(complex64 或者 complex128)。

参数

  • x (Tensor) - 输入 Tensor

返回

bool,如果输入 tensor 的数据类型为复数类型则为 True,反之为 False

代码示例

>>> import paddle

>>> x = paddle.to_tensor([1 + 2j, 3 + 4j])
>>> print(paddle.is_complex(x))
True

>>> x = paddle.to_tensor([1.1, 1.2])
>>> print(paddle.is_complex(x))
False

>>> x = paddle.to_tensor([1, 2, 3])
>>> print(paddle.is_complex(x))
False