I have a very simple usecase that seems to be way too difficult in mxnet. This is my input: x = mx.nd.array([[ 1., 2., 3., 4.], [ 5., 6., 7., 8.], [ 9., 10, 11, 12]]) I want to select/pick/take/whatever items on each row. For example on the first row the first 2 elements, then on the second …

Simple NDArray/Symbol indexing

sad September 19, 2019, 7:00pm 3

I have answer here that explains the difference between pick and take as well:

Topic		Replies	Views
Pick vs Take	1	666	August 19, 2019
How to choose some specific rows according to 0/1 mask?	4	610	October 21, 2019
Workaround for take operation on other axes Discussion	1	388	June 29, 2018
Ndarray problem Gluon	1	423	September 19, 2019
Array newbie Discussion	0	294	February 21, 2022