triton.language.broadcast

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

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

参数:

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

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