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

68 votes172 views6 answers
Python dict sort by value
Python
0 vote182 views1 answer
Generate 3 random integers between 100 and 999 which is divisible by 5
Python
0 vote167 views1 answer
Example 1: Reset Index & Drop Old Index pandas
Python
5 votes191 views2 answers
Python get min max value from a dictionary
Python
1 vote219 views1 answer
Import "django.core.urlresolvers" could not be resolved
Python Django
0 vote176 views1 answer
Iniciar servidor en Django
Python
0 vote305 views1 answer
How to use sin inverse and cos inverse in python
Python
0 vote184 views1 answer
Python find HCF
Python
2 votes192 views1 answer
W=how to tell if decimal in python
Python
1 vote198 views1 answer
Get n items from dictionary python
Python