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 vote360 views1 answer
Kmpm]pomfyukruk6nfgngnzgnzggngnxfgnfgxfgfgxfggnxfggngnggngngngngngngngn. ';';';';'; ; ; ; ;; ; ; ; ; ; ; ; ;
Python
0 vote370 views2 answers
Most common letter in string python
Python
0 vote358 views1 answer
Django on_delete rules
Python
2 votes385 views1 answer
C:\Users\saverma2>notebook 'notebook' is not recognized as an internal or external command, operable program or batch file.
Python
1 vote343 views1 answer
Guido van rossum net worth
Python
0 vote343 views1 answer
Odd or even checker
Python
0 vote361 views1 answer
Get timestamp with pyrhon
Python
0 vote353 views1 answer
Codes for infrntial statistics in python
Python
0 vote345 views1 answer
No module named 'sqlalchemy' mac
Python
0 vote348 views1 answer
Which type of programming does python support?
Python