IndexSelect =========== Returns a new tensor which indexes the input tensor along dimension dim using the entries in index. **Abstract Signature:** ``IndexSelect(input: Tensor, dim: int, index: Tensor)`` .. raw:: html
torch.index_select
jax.numpy.take
numpy.take
keras.ops.take
tf.gather
mlx.core.take