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 votes368 views1 answer
Django get current date
Python Django
0 vote361 views1 answer
How to get parent model object based on child model filter in django
Python Django
0 vote337 views1 answer
Listing of django model types
Python
3 votes337 views1 answer
How to set datetime format in python
Python
1 vote367 views1 answer
How to open chrome console in selenium webdriver
Python
0 vote362 views1 answer
For loop in django template css
Python Django
1 vote301 views1 answer
Drop rows pandas where value
Python
1 vote258 views1 answer
Python if boolean
Python
0 vote273 views1 answer
Hackerrank solutions python 3
Python
0 vote312 views1 answer
How to print out even index in python
Python