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.

param a:

Tile to broadcast

param shape:

The shape to broadcast to

returns:

Tile with broadcast shape