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 vote212 views1 answer
Multi-line comments Python Example
Python
0 vote206 views1 answer
Read_csv only certain columns
Python
4 votes197 views2 answers
Python list comprehension double for
Python
3 votes224 views6 answers
Send embed discord.py
Python
0 vote208 views1 answer
Set environment variable flask app
Python
0 vote201 views1 answer
Merge multile dict
Python
0 vote232 views1 answer
Get n largest values from 2D numpy array matrix
Python
0 vote210 views1 answer
Python plot lines with dots
Python
15 votes201 views6 answers
Python sqlite3
Python
1 vote227 views1 answer
Updating many to many field in django using pk
Python Django