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]]