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 vote111 views1 answer
Elif in django template
Python
0 vote133 views1 answer
Python change terminal name
Python
0 vote118 views1 answer
Using strip() method to remove the newline character from a string
Python
2 votes106 views1 answer
Who is a pythonista
Python
2 votes97 views1 answer
What skills do you need to master pvp in minecraft
Python
0 vote120 views1 answer
Check if cuda is present in torch
Python
4 votes97 views1 answer
Round python with list
Python
7 votes93 views1 answer
Python sqrt import
Python
2 votes250 views1 answer
Two dimensional array python
Python
1 vote105 views2 answers
Python not null
Python