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

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