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

2 votes262 views3 answers
Get webpage data as json python
Python
10 votes259 views2 answers
Check if key exists in sesison python
Python
0 vote283 views1 answer
Parquet to dataframe
Python
0 vote305 views1 answer
Python, read_csv from string
Python
-1 vote275 views1 answer
Harihar kaka class 10 questions
Python
0 vote266 views1 answer
FIND MISSING NUMBER IN AN ARRAY IN PYTHON
Python
1 vote316 views1 answer
Discord.py unmute
Python
2 votes397 views2 answers
TypeError: dict is not a sequence
Python
2 votes261 views6 answers
Update ubuntu to python 3.85
Python
0 vote265 views2 answers
Stackoverflow search
Python