triton.language.broadcast

triton.language.broadcast(input, other)

Tries to broadcast the two given blocks to a common compatible shape.

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

  • other (Block) – The second input tensor.