//clika-runtime/io.clika.runtime/Ops/indexSelect

indexSelect

[common]
fun indexSelect(input: Tensor, dim: Long, index: Tensor): Tensor

indexSelect(input: Tensor, dim: Long, index: Tensor): the index_select operator. Select whole rows of input at index positions along dim. Output = input's shape with the dim extent replaced by index's length; rows may repeat and appear in any order. index is 1-D Int32/Int64; prefer Int32 where the extents allow (half the index bytes).