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 vote146 views3 answers
Remove all parentheses from string python
Python
1 vote135 views1 answer
How to check if two columns match in pandas
Python
0 vote182 views1 answer
Python wait for executable to finish before perceeding
Python
1 vote134 views1 answer
How to clear a pickle file
Python
4 votes148 views1 answer
How to round the values in a list
Python
0 vote165 views1 answer
Extract tgz files in python
Python
0 vote136 views1 answer
How to access dataframe row by datetime index
Python
15 votes213 views4 answers
Django python install
Python Django
-2 votes154 views1 answer
Requests.exceptions.connectionerror: ('connection aborted.', remotedisconnected('remote end closed connection without response'))
Python
0 vote218 views1 answer
Django float validator
Python Django