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

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。非经特殊声明，原始代码版权归原作者所有，本译文未经允许或授权，请勿转载或复制。