triton.language.cat

triton.language.cat(input, other, can_reorder=False, dim=0, _semantic=None)

Concatenate the given blocks

Parameters:

• input (Tensor) – The first input tensor.

• other (Tensor) – The second input tensor.

• can_reorder (bool) – Compiler hint. If true, the compiler is allowed to reorder elements while concatenating inputs. Only use if the order does not matter (e.g., result is only used in reduction ops).

• dim (int) – The dimension to concatenate along (used when can_reorder is False).