triton.language.interleave

triton.language.interleave(a, b)

沿最后一个维度交错两个张量的值。两个张量必须具有相同的形状。等同于 tl.join(a, b).reshape(a.shape[:-1] + [2 * a.shape[-1]])

参数：

• a (Tensor) – 第一个输入张量。

• b (Tensor) – 第二个输入张量。