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 votes245 views4 answers
ValueError: Input contains NaN, infinity or a value too large for dtype('float32'). site:stackoverflow.com
Python
0 vote233 views1 answer
Pyqt stretch image
Python
14 votes234 views3 answers
Getting ip address based on phone number python
Python
0 vote240 views1 answer
Compile python folder
Python
0 vote274 views1 answer
Pyspark name accumulator
Python
7 votes240 views1 answer
Smtp email template
Python
0 vote243 views3 answers
“no such column” after adding a field to the model
Python
0 vote245 views1 answer
Python returning rows and columns from a matrix string
Python
0 vote243 views1 answer
Drop row with duplicate value
Python
0 vote246 views1 answer
Access CSV Files in Google Colab
Python