numpy has a way to select elements use bool ndarray with same shape. like this:
x = np.arange(1 * 2 * 3).reshape((1, 2, 3))
x[x >= 3]
output: array([3, 4, 5])
but mxnet can’t do this… (I also tried pick(), scatter_nd(), take())
numpy has a way to select elements use bool ndarray with same shape. like this:
x = np.arange(1 * 2 * 3).reshape((1, 2, 3))
x[x >= 3]
output: array([3, 4, 5])
but mxnet can’t do this… (I also tried pick(), scatter_nd(), take())