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

-4 votes150 views3 answers
Python read all text files in directory
Python
3 votes139 views2 answers
Godot restart scene
Python
0 vote125 views1 answer
NameError: name 'accuracy_score' is not defined
Python
0 vote127 views1 answer
How to define piecewise function i python
Python
4 votes116 views1 answer
Python data structures
Python
0 vote126 views1 answer
ModuleNotFoundError: No module named 'nbformat'
Python
2 votes141 views2 answers
Python - Count the Number of Keys in a Python Dictionary
Python
1 vote88 views1 answer
Typeerror: list indices must be integers or slices, not str
Python
0 vote131 views2 answers
Python which packages depend on package
Python
1 vote141 views2 answers
Assign multiline string to variable in python
Python