triton.language.interleave

triton.language.interleave(a, b)

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

Parameters:

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

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