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 vote156 views3 answers
Plt plot circle
Python
1 vote145 views1 answer
Python requests.get timeout
Python
1 vote199 views2 answers
Python int16
Python
1 vote157 views1 answer
Wheter
Python
1 vote149 views1 answer
Python order 2d array by secode element
Python
0 vote151 views1 answer
Python sort by value first then key lexicography
Python
92 votes155 views3 answers
Renaming column in dataframe pandas
Python
0 vote176 views1 answer
Python inject into process
Python
1 vote164 views1 answer
Pros and cons of python flush print function
Python
5 votes161 views3 answers
Correlation matrix python
Python