當前位置: 首頁>>代碼示例 >>用法及示例精選 >>正文

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。非經特殊聲明，原始代碼版權歸原作者所有，本譯文未經允許或授權，請勿轉載或複製。