mindspore.mint.isneginf

查看源文件
mindspore.mint.isneginf(input)[源代码]

确定输入Tensor每个位置上的元素是否为负无穷。

警告

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

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

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

返回:

Tensor,与输入具有相同形状,其中元素在对应输入为负无穷大时为 True,否则为 False

异常:
  • TypeError - 如果 input 不是Tensor。

支持平台:

Ascend

样例:

>>> from mindspore import mint, Tensor
>>> from mindspore import dtype as mstype
>>> output = mint.isneginf(Tensor([[-float("inf"), float("inf")], [1, -float("inf")]], mstype.float32))
>>> print(output)
[[ True False]
 [False  True]]