triton.language.broadcast_to

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

Tries to broadcast the given block to a new shape.

Parameters

• input (Block) – The input block.

• shape (Tuple[int]) – The desired shape.

• builder (triton.ir.builder, optional from within JIT'ed functions) – IR builder to generate code into