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

12 votes279 views4 answers
Remove nan particular column pandas
Python
4 votes288 views2 answers
Change django admin title
Python
31 votes262 views4 answers
Convert excel to csv python program
Python
9 votes329 views2 answers
Python check if number is in range
Python
17 votes287 views3 answers
Set title matplotlib
Python
0 vote293 views1 answer
Yum install python3
Python
0 vote264 views1 answer
Python discover methods of object/module
Python
0 vote272 views1 answer
How to split a string with newline in python
Python
2 votes294 views6 answers
Scikit learn
Python
0 vote269 views1 answer
Celery subprocess
Python