Numpy multidimensional indexing

# Let's say you have a batch of 256 inputs, each one has a set of 64 points.
x.shape --> (256, 64, 3)

# Now assume you want to index each image and draw 128 samples of those points,
# with an indices matrix.
idx.shape --> (256, 128)

# Do
y = x[np.arange(x.shape[0]), idx.T].reshape((1, 0, 2))
y.shape --> (256, 128, 3)

Similar codes

2 votes301 views10 answers
Lol
Python
1 vote314 views1 answer
Cv2 draw box
Python
-4 votes320 views4 answers
Python 3.4 release date
Python
0 vote307 views1 answer
How to send more than one variables to python using xlwings
Python
1 vote291 views1 answer
Count how many times a value shows in python list
Python
0 vote302 views2 answers
Arma-garch python
Python
8 votes294 views2 answers
How to read text frome another file pythion
Python
33 votes299 views10 answers
Elon musk
Python
3 votes318 views2 answers
Django radio button
Python
103 votes600 views6 answers
Time.sleep() python
Python