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 vote163 views1 answer
How to make python turn a list into a text file grapper
Python
-1 vote172 views2 answers
How to find a square root of a number using python
Python
2 votes166 views3 answers
Pandas count occurrences of certain value in row
Python
0 vote213 views1 answer
Delete all historical data django simple history
Python Django
2 votes160 views1 answer
Django load model by name
Python
0 vote157 views3 answers
Join list with comma python
Python
1 vote188 views1 answer
Streamlit button to load a file
Python
5 votes191 views1 answer
How to get all links from a website python beautifulsoup
Python
2 votes165 views5 answers
Python xml parser
Python
0 vote183 views1 answer
Python import environment variables
Python