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 vote281 views1 answer
"DO_NOTHING" is not defined django
Python
1 vote302 views2 answers
Block window if another window is open tkinter
Python
16 votes276 views6 answers
Map in python
Python
0 vote348 views1 answer
Decoding
Python
0 vote272 views1 answer
Lag function in pandas
Python
0 vote287 views1 answer
Default python structure
Python
0 vote288 views1 answer
Reassign variable python
Python
0 vote284 views1 answer
A int and float. python
Python
0 vote288 views2 answers
Pandas sub dataframe
Python
0 vote359 views2 answers
Flask lazy response style with `make_response`
Python Flask