triton.broadcast_to

triton.broadcast_to(input, shape, builder=None)

Tries to broadcast a block to a new shape.

Parameters

• input (triton.value) – The input block.

• shape (tuple of int) – The new shape.

• builder (triton.ir.builder) – IR builder to generate code into, optional from within @triton.jit functions