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

20 votes265 views5 answers
Save list to file python
Python
0 vote264 views1 answer
Django queryset with multiple contain keyword
Python
0 vote237 views1 answer
How to log errors while debug is false in django
Python
0 vote265 views1 answer
Python sort class by attribute
Python
25 votes247 views2 answers
How to address null in python
Python
7 votes247 views3 answers
Make exe file pycharm
Python
0 vote238 views1 answer
How to store something in python
Python
20 votes293 views4 answers
Python pandas read_excel
Python
0 vote281 views1 answer
Python - columns that contain the lengh of a string
Python
0 vote295 views1 answer
Python package for misspelled words
Python