triton.language.broadcast

triton.language.broadcast(input, other)

尝试将给定的两个块广播到一个共同的兼容形状。

Parameters:

• input (Block) – 第一个输入张量。

• other (Block) – 第二个输入张量。