triton.language.cat

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

连接给定的数据块

参数:

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

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

• can_reorder (bool) – 编译器提示。如果为真，编译器可以在连接输入时重新排列元素。仅在顺序无关紧要时使用（例如，结果仅用于归约操作）。

• dim (int) – 进行连接的维度（当 can_reorder 为 False 时使用）。