View page source

warp.tile_slice_indexed#

warp.tile_slice_indexed(
t: Tile[Any, tuple[int, ...]],
indices: tuple,
) → Tile[Any, tuple[int, ...]]#

  • Kernel

  • Differentiable

Gather elements of a tile along a single axis using a 1D tile of integer indices.

This lowers the advanced-indexing syntax t[indices, :], gathering elements of t along one axis given by indices. All other axes must be selected in full.

Parameters:

  • t – Input tile to gather from

  • indices – A 1D tile of integer indices selecting elements along one axis

Returns:

A register tile whose extent along the indexed axis equals the number of indices.