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

-1 vote278 views1 answer
Find the range in python
Python
10 votes247 views4 answers
Pandas find column with max value for each row
Python
0 vote277 views1 answer
Python : read all the contents of the file into a string (use of 'with open')
Python
17 votes262 views5 answers
Remove punctuation python
Python
2 votes260 views2 answers
Trim starting space python
Python
0 vote259 views1 answer
Python discord if == emoji not working
Python
-1 vote253 views1 answer
Receipt parsing
Python
0 vote273 views1 answer
Pandas string manipulation on column
Python
0 vote244 views1 answer
Highlight Active Links in Django Website
Python
0 vote280 views1 answer
Boxplot for all columns in python
Python