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

13 votes163 views5 answers
How to check if a input is an integer python
Python
0 vote147 views10 answers
Python generator cheat sheet download
Python
0 vote159 views1 answer
Months of the year python list
Python
0 vote136 views1 answer
Django app directory
Python
0 vote155 views1 answer
Python cast number to between 0 and 1
Python
0 vote153 views1 answer
Np.loadtext
Python
0 vote140 views3 answers
Webbrowser.google.open python
Python
22 votes145 views6 answers
Tkinter button
Python
0 vote174 views1 answer
Pd.DataFrame(lm.coef_,X.columns,columns=['Coeff'])
Python
0 vote193 views2 answers
Style django forms with crisp
Python Django