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

3 votes132 views3 answers
Range(n,n) python
Python
0 vote146 views1 answer
Python sleeping with a varible
Python
0 vote182 views1 answer
Sns save chart
Python
40 votes145 views1 answer
Convert string to list python
Python
3 votes138 views2 answers
AttributeError: module 'tensorflow' has no attribute 'GraphDef'
Python
13 votes194 views2 answers
Delete database entry using name django
Python Django
0 vote152 views1 answer
Stemming words python
Python
0 vote146 views1 answer
Picture plot
Python
11 votes168 views6 answers
Curl to python
Python
32 votes138 views6 answers
Array in python
Python