mindspore.ops.isinf

mindspore.ops.isinf(input)

返回一个布尔型tensor，指示哪些元素是正负无穷数。

警告

• 这是一个实验性API，后续可能修改或删除。

• 该API目前只支持在Atlas A2训练系列产品上使用。

参数：

• input (Tensor) - 输入tensor。

返回：

Tensor

支持平台：

Ascend CPU GPU

样例：

>>> import mindspore
>>> input = mindspore.tensor([-1, 3, float("inf"), float("-inf"), float("nan")])
>>> mindspore.ops.isinf(input)
Tensor(shape=[5], dtype=Bool, value= [False, False,  True,  True, False])