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.