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 vote127 views1 answer
Maximum number of guests on cruise at an instance tcs
Python
0 vote109 views1 answer
How to take a list as input in python using sys.srgv
Python
2 votes121 views1 answer
Timeout exception in selenium python
Python
0 vote120 views1 answer
Django active link
Python Django
10 votes114 views3 answers
How to make a rect in pygame
Python
0 vote109 views2 answers
Python selenium switch to window
Python
37 votes110 views6 answers
Json dumps python
Python
0 vote71 views1 answer
Python to datetime day of year
Python
0 vote112 views1 answer
Timestamp' object has no attribute 'isnull
Python
2 votes108 views1 answer
How do i print a list line by line in python
Python