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


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


用法:

mxnet.ndarray.contrib.isfinite(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.isfinite(data)
>>> output
[0. 0. 0. 1.]
<NDArray 4 @cpu(0)>

相关用法


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