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 vote142 views1 answer
Fetch row where column is missing pandas
Python
0 vote122 views1 answer
Cannot cast type smallint to boolean django
Python
0 vote118 views1 answer
How to change pi hostname in python file
Python
0 vote128 views1 answer
Doublespace in python
Python
0 vote155 views1 answer
How to add column to heroku postgres in my django app
Python Django
57 votes134 views6 answers
Train test split sklearn
Python
168 votes206 views6 answers
How to upgrade pip in python
Python
7 votes128 views4 answers
Python tkinter button multiple commands
Python
0 vote128 views1 answer
Standard noramlization
Python
14 votes137 views1 answer
Sort list of dictionaries by key python
Python