warp.tile_broadcast#

warp.tile_broadcast(
a: Tile[Any, tuple[int, ...]],
shape: tuple[int, ...],
) → Tile[Any, tuple[int, ...]]#

  • Kernel

  • Differentiable

Broadcast a tile.

Broadcasts the input tile a to the destination shape. Broadcasting follows NumPy broadcast rules.

Parameters:

  • a – Tile to broadcast

  • shape – The shape to broadcast to

Returns:

Tile with broadcast shape.