triton.language.cat

triton.language.cat(input, other, can_reorder=False)

连接给定的块

Parameters:

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

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

• reorder - 编译器提示。如果为true，编译器在连接输入时允许重新排序元素。仅当顺序无关紧要时使用（例如，结果仅用于归约操作）。当前cat的实现仅支持can_reorder=True。