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 vote237 views1 answer
Python dict to url params
Python
0 vote222 views1 answer
Pandas from range of columns
Python
3 votes231 views1 answer
How does scope work in python
Python
-1 vote189 views1 answer
Python fiboncci
Python
0 vote216 views1 answer
Break line in string python
Python
0 vote179 views1 answer
Python draw state diagrams
Python
8 votes244 views2 answers
Pandas iterate columns
Python
1 vote215 views1 answer
Python how to use a variable to trigger an event
Python
1 vote248 views2 answers
Join two querysets django
Python
1 vote241 views1 answer
Python list all csv in dir
Python