warp.tile_squeeze#

warp.tile_squeeze(
t: Tile[Any, tuple[int, ...]],
axis: tuple[int, ...],
) → Tile[Any, tuple[int, ...]]#

  • Kernel

  • Differentiable

Create a squeezed view of a tile with the same data.

Parameters:

  • t – Input tile to squeeze

  • axis – A subset of the entries of length one in the shape (optional)

Returns:

The input tile but with all or a subset of the dimensions of length one removed.