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

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

用法:

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

參數：

  • data：(Symbol) - 數據
  • index：(Symbol) - 掩碼
  • axis：(int, optional, default='0') - 一個整數，表示 NDArray 中要屏蔽的軸。
  • name：(string, optional.) - 結果符號的名稱。

返回：

結果符號。

返回類型：

Symbol

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