triton.language.broadcast_to

triton.language.broadcast_to(input, shape)

Tries to broadcast the given tensor to a new shape.

Parameters:
  • input (Block) – The input tensor.

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