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


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


用法:

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