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]])