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 vote240 views1 answer
How to list gym envirolments
Python
1 vote193 views1 answer
Accessing index of dataframe python
Python
-2 votes187 views1 answer
Numpy difference between two arrays
Python
0 vote175 views1 answer
Python turtle set direction
Python
0 vote176 views1 answer
Example 1: Reset Index & Drop Old Index pandas
Python
0 vote221 views1 answer
Generate python
Python
1 vote202 views1 answer
Postgres django settings
Python
0 vote191 views1 answer
Columns overlap but no suffix specified: Index(['zpid'], dtype='object')
Python
0 vote202 views1 answer
Pyqt5 python how to add a widgtet to a table
Python
0 vote190 views1 answer
Python savefig full screen
Python