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

0 vote300 views1 answer
Align a text python
Python
0 vote311 views1 answer
Python wget anaconda
Python
13 votes320 views6 answers
Range in python
Python
2 votes327 views1 answer
Mario dance dance revolution
Python
0 vote386 views2 answers
Flask get summernote text
Python Flask
39 votes341 views4 answers
No module named 'matplotlib'
Python
0 vote329 views1 answer
Inherit functions from other classes
Python
13 votes359 views4 answers
Python calculate age from date of birth
Python
0 vote304 views1 answer
Pandas currency to numbe
Python
0 vote322 views1 answer
Assign more than one variable at a time on a single line in python
Python