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

Python mxnet.ndarray.contrib.boolean_mask用法及代碼示例


用法:

mxnet.ndarray.contrib.boolean_mask(data=None, index=None, axis=_Null, out=None, name=None, **kwargs)

參數：

  • data：(NDArray) - 數據
  • index：(NDArray) - 掩碼
  • axis：(int, optional, default='0') - 一個整數，表示 NDArray 中要屏蔽的軸。
  • out：(NDArray, optional) - 輸出 NDArray 來保存結果。

返回：

out：- 此函數的輸出。

返回類型：

NDArray 或 NDArray 列表

給定 n-d NDArray 數據和一維 NDArray 索引，運算符生成 un-predeterminable 形狀的 n-d NDArray 輸出，它代表 x 中索引中對應元素非零的行。

>>> data = mx.nd.array([[1, 2, 3],[4, 5, 6],[7, 8, 9]])
>>> index = mx.nd.array([0, 1, 0])
>>> out = mx.nd.contrib.boolean_mask(data, index)
>>> out

[[4。 5. 6.]] <NDArray 1x3 @cpu(0)>

相關用法


注：本文由純淨天空篩選整理自apache.org大神的英文原創作品 mxnet.ndarray.contrib.boolean_mask。非經特殊聲明，原始代碼版權歸原作者所有，本譯文未經允許或授權，請勿轉載或複製。