请问未来NDarray会支持使用boolean mask的indexing吗?

Ndarray进行逻辑运算后就是返回boolean mask却没办法用它取元素
最近在实现网络的时候发现没有这个的话会很麻烦,有时候必须要转成Numpy去做
https://docs.scipy.org/doc/numpy-1.13.0/user/basics.indexing.html#boolean-or-mask-index-arrays

以上的讨论是基于:
MXNet版本: 1.2.0

1赞

目前也遇到这个问题,请问有什么好的方法吗?

有解决方法,麻烦了点,而且有一个很重要的前提条件:
你必须知道要取元素的数量,或者还是原来的数量其它用默认值填充

然后就是用mx.nd.where方法获取元素:

比如:
nd.where(mask, value, dummy_value)

1赞

可以用 mxnet.contrib.ndarray.boolean_mask()

$$fdfdf$$