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

2 votes138 views2 answers
How to play audio in python
Python
5 votes210 views1 answer
Convert a data frame column values to list
Python
1 vote186 views1 answer
Create django group
Python
5 votes198 views2 answers
Django password field
Python Django
0 vote174 views2 answers
Axes color python
Python
0 vote178 views3 answers
Pip proxy settings
Python
8 votes233 views2 answers
Pandas dataframe get number of columns
Python
3 votes184 views1 answer
Blinking an led with raspberry pi
Python
0 vote185 views1 answer
Exercise of python loops
Python
0 vote178 views1 answer
Positive counter python
Python