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 vote155 views1 answer
Fibonacci sequence in python 2.7
Python
0 vote160 views1 answer
Plt title color
Python
4 votes184 views1 answer
Python: change column name
Python
0 vote173 views1 answer
Python counting subfolders on specific level
Python
0 vote159 views1 answer
Python. printing varibles
Python
0 vote175 views1 answer
Python scipy put more weight to a set value in curve_fit
Python
0 vote182 views1 answer
Import data in pandad
Python
1 vote181 views1 answer
Python number divisible by two other numbers
Python
-1 vote191 views1 answer
Django model UUID field
Python Django
0 vote171 views1 answer
1250 minutes to hours
Python