triton.language.broadcast_to(input, *shape)

Tries to broadcast the given tensor to a new shape.

  • input (Block) – The input tensor.

  • shape – The desired shape.

shape can be passed as a tuple or as individual parameters:

# These are equivalent
broadcast_to(x, (32, 32))
broadcast_to(x, 32, 32)

This function can also be called as a member function on tensor, as x.broadcast_to(...) instead of broadcast_to(x, ...).