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 vote207 views1 answer
Gui button in tkinter color
Python
4 votes284 views3 answers
How to loop through dates in python
Python
1 vote255 views1 answer
__delattr__ python
Python
3 votes193 views1 answer
Python reciprocal
Python
17 votes213 views6 answers
Euclidean distance python
Python
2 votes232 views1 answer
Django register models
Python Django
6 votes187 views2 answers
If file exist in folder then delete in python \
Python
0 vote185 views1 answer
Python pandas get labels
Python
0 vote204 views1 answer
Create contract from interface in brownie
Python
4 votes194 views4 answers
Urllib python
Python