mindspore.ops.is_nonzero

mindspore.ops.is_nonzero(input)[源代码]

判断输入Tensor是否包含0或False,输入只能是单元素。

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

返回:

Bool,如果输入Tensor包含一个单元素0或者一个单元素False,则返回False,否则返回True。

异常:
  • TypeError - input 不是Tensor。

  • ValueError - input 的元素数量不等于1。

支持平台:

Ascend GPU CPU

样例:

>>> from mindspore import Tensor, ops
>>> x1 = Tensor([[[False]]])
>>> x2 = Tensor([[3.5]])
>>> out1 = ops.is_nonzero(x1)
>>> print(out1)
False
>>> out2 = ops.is_nonzero(x2)
>>> print(out2)
True