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

0 vote152 views1 answer
Discord embed add image
Python
0 vote168 views1 answer
Django allauth Reverse for 'password_reset_confirm' not found. 'password_reset_confirm' is not a valid view function or pattern name.
Python
0 vote168 views2 answers
Data frame list value change to string
Python
2 votes150 views2 answers
Python Decompress gzip File
Python
2 votes134 views2 answers
Php echo shorthand
Python
0 vote218 views1 answer
Gpt2 simple restore_from
Python
0 vote169 views1 answer
The grandest staircase of them all foobar solution
Python
0 vote147 views1 answer
Tkinter sin
Python
2 votes147 views1 answer
Comparing file content in python
Python
0 vote157 views2 answers
Tsv to csv python
Python