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 vote264 views1 answer
Get nonzero min numpy
Python
0 vote260 views1 answer
Try python import
Python
13 votes194 views2 answers
How to convert python input to int
Python
-1 vote196 views1 answer
Pathlib change extension
Python
0 vote140 views1 answer
Unhapppy man with monwy
Python
0 vote189 views1 answer
Inspectdb django
Python
10 votes241 views2 answers
Selenium refresh page python
Python
0 vote191 views1 answer
Dynamic grid using tkinter
Python
15 votes195 views6 answers
Django models
Python
0 vote196 views1 answer
Weight constraints keras cnn
Python