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

1 vote257 views1 answer
Python dont exit script after print
Python
10 votes176 views2 answers
Python get numbers after decimal point
Python
0 vote178 views1 answer
Accuracy for each class
Python
2 votes198 views1 answer
Values of unique from dataframe with count
Python
0 vote132 views1 answer
Create series with number intervals
Python
1 vote182 views3 answers
Program to draw rectangle in python
Python
4 votes241 views2 answers
Primary key django model
Python Django
18 votes166 views4 answers
Reset index python
Python
1 vote180 views1 answer
Python parallel list comprehension
Python
0 vote205 views1 answer
RuntimeError: Expected object of type torch.FloatTensor but found type torch.cuda.FloatTensor for argument
Python