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 votes154 views2 answers
Extract one column from dataframe python
Python
2 votes149 views2 answers
Check multiple keys in python dict
Python
103 votes171 views6 answers
Time.sleep() python
Python
10 votes153 views5 answers
Check gpu in tensorflow
Python
0 vote148 views1 answer
Django auth user model
Python
0 vote137 views1 answer
Download textdocuments with python
Python
3 votes174 views1 answer
Write binary file in python
Python
2 votes184 views1 answer
Python ftplib get list of directories
Python
1 vote137 views1 answer
Staticfiles
Python
0 vote110 views1 answer
Python ansi color codes
Python