mindspore.ops.is_complex

查看源文件
mindspore.ops.is_complex(input)[源代码]

返回一个布尔型tensor,指示哪些元素是复数。

参数:
  • input (Tensor) - 输入tensor。

返回:

Bool

支持平台:

Ascend GPU CPU

样例:

>>> import mindspore
>>> input = mindspore.tensor([1, 1+1j, 2+2j], mindspore.dtype.complex64)
>>> mindspore.ops.is_complex(input)
True
>>>
>>> input = mindspore.tensor([1, 1+1j, 2+2j], mindspore.dtype.complex128)
>>> mindspore.ops.is_complex(input)
True
>>> input = mindspore.tensor([1, 1+1j, 2+2j], mindspore.dtype.int32)
>>> mindspore.ops.is_complex(input)
False