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

7 votes113 views1 answer
Local image embed discord py
Python
34 votes106 views3 answers
Download dataframe as csv
Python
1 vote118 views1 answer
Python convert to percentage
Python
1 vote116 views1 answer
Unix command in python script
Python
0 vote128 views1 answer
How to do formatting in python with format function
Python
15 votes167 views6 answers
Matplotlib.pyplot
Python
1 vote116 views1 answer
How to veiw and edit files with python
Python
1 vote114 views2 answers
Csv reader python skip header
Python
8 votes110 views3 answers
How to delete nan values in python
Python
0 vote101 views1 answer
Customise the django rest api view
Python