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

11 votes217 views4 answers
Python get random character from string
Python
15 votes250 views2 answers
Python from timestamp to string
Python
0 vote240 views1 answer
Python if dataframe has at least one row
Python
5 votes232 views2 answers
Converting decimal to hex in python
Python
6 votes209 views2 answers
Numpy ones
Python
1 vote233 views1 answer
Python pandas dataframe column date to string
Python
0 vote242 views1 answer
See python function details in vscode
Python
0 vote223 views2 answers
Arg parse array argument
Python
11 votes202 views6 answers
If statement in python
Python
0 vote263 views1 answer
Decoding
Python