triton.language.broadcast

triton.language.broadcast(input, other, _semantic=None)

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

参数:

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

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