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 vote130 views1 answer
Django choicefield empty label
Python
0 vote134 views1 answer
Assignment 6.5 python for everybody
Python
0 vote130 views1 answer
Make exact copy of an environment
Python
3 votes153 views1 answer
How to append to front of the list using def method in python 3
Python
0 vote145 views1 answer
Python string to list new line
Python
1 vote161 views2 answers
Python alphabetical order
Python
1 vote137 views2 answers
React icon size
Python
0 vote218 views1 answer
How to make my discord bot shut down with a command
Python
1 vote142 views1 answer
How to set up dataframe from csv
Python
22 votes137 views3 answers
How to call a random function in python
Python