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 votes315 views4 answers
Python find if string contains space
Python
0 vote337 views1 answer
List get both index and value.
Python
-1 vote332 views2 answers
Matplotlib set colorbar range
Python
0 vote347 views2 answers
Cumulative chart python plotly
Python
6 votes372 views2 answers
Get client ip flask
Python Flask
10 votes343 views4 answers
Count values pandas
Python
0 vote325 views1 answer
Python remove .0
Python
1 vote340 views1 answer
Insert missing data in pandas
Python
1 vote336 views1 answer
Yapf ignore line
Python
0 vote331 views1 answer
Split at the second occurrence of the element python
Python