当前位置: 首页>>代码示例 >>用法及示例精选 >>正文


Python mxnet.ndarray.contrib.isinf用法及代码示例


用法:

mxnet.ndarray.contrib.isinf(data)

参数

input(NDArray) - N-D NDArray。

返回

output- 输出 NDarray,形状与输入相同,其中 1 表示数组元素等于正无穷或负无穷,否则为 0。

返回类型

ND阵列

执行逐元素检查以确定 NDArray 是否包含无限元素。

例子

>>> data = mx.nd.array([np.inf, -np.inf, np.NINF, -1])
>>> output = mx.nd.contrib.isinf(data)
>>> output
[1. 1. 1. 0.]
<NDArray 4 @cpu(0)>

相关用法


注:本文由纯净天空筛选整理自apache.org大神的英文原创作品 mxnet.ndarray.contrib.isinf。非经特殊声明,原始代码版权归原作者所有,本译文未经允许或授权,请勿转载或复制。