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 vote203 views2 answers
How to convert time from one timezone to another in python
Python
2 votes219 views1 answer
Python append to first index
Python
2 votes186 views1 answer
What is the difference between the use of iloc and loc?
Python
0 vote206 views1 answer
Micropython string to int
Python
1 vote220 views1 answer
How to get the value out of a dictionary python3
Python
0 vote227 views1 answer
Torch print floating precision
Python
2 votes208 views1 answer
Python backslash in string
Python
1 vote245 views1 answer
Sum values in django models
Python
3 votes229 views1 answer
Send email hotmail using python
Python
0 vote226 views1 answer
Turn list of arrays into array
Python