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 vote368 views1 answer
Root = tk() python 3
Python
0 vote395 views1 answer
Django form label in template
Python Django
0 vote361 views1 answer
IBM Maximo automation python
Python
114 votes385 views5 answers
Change name of column pandas
Python
0 vote349 views1 answer
Dir template
Python
4 votes321 views2 answers
How to count range in django template
Python
0 vote321 views1 answer
Send command dynamo civid
Python
8 votes345 views3 answers
F.readlines python
Python
0 vote388 views1 answer
Python format subprocess output
Python
0 vote345 views1 answer
Log transform pandas dataframe
Python