mindspore.numpy.isinf

查看源文件
mindspore.numpy.isinf(x, dtype=None)[源代码]

逐元素测试是否为正无穷大或负无穷大。 返回一个与 x 具有相同shape的bool数组,当 x == +/-inf 时为True,否则为False。

说明

不支持Numpy的 outwherecastingordersuboksignatureextobj 参数。目前仅支持 np.float32

参数:
  • x (Tensor) - 输入值。

  • dtype (mindspore.dtype, 可选) - 默认值: None 。覆盖输出Tensor的dtype。

返回:

Tensor或标量,当 x 为正无穷大或负无穷大时为True,否则为False。如果 x 是标量,则返回标量。

支持平台:

GPU CPU

样例:

>>> import mindspore.numpy as np
>>> output = np.isinf(np.array(np.inf, np.float32))
>>> print(output)
True
>>> output = np.isinf(np.array([np.inf, -np.inf, 1.0, np.nan], np.float32))
>>> print(output)
[ True  True False False]