triton.language.interleave

triton.language.interleave(a, b)

Interleaves the values of two tensors along their last dimension. The two tensors must have the same shape. Equivalent to tl.join(a, b).reshape(a.shape[-1:] + [2 * a.shape[-1]])

Parameters:

• a (Tensor) – The first input tensor.

• b (Tensor) – The second input tensor.