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 vote106 views1 answer
Valueerror need more than 2 values to unpack findcontours
Python
2 votes103 views1 answer
Check whether number is even or odd
Python
0 vote134 views1 answer
Show post id on django admin interface
Python Django
1 vote147 views4 answers
Join tables in django orm
Python Django
0 vote106 views1 answer
How to find permutation of numbers in python
Python
0 vote143 views2 answers
Get lastest files from directory python
Python
1 vote111 views1 answer
Cannot import name 'imputer'
Python
7 votes113 views3 answers
Determine if number is prime python
Python
1 vote107 views1 answer
Clamp number in python
Python
7 votes111 views1 answer
Get inverse dict python
Python