【Triton 教程】triton_language.cat

Triton 是一种用于并行编程的语言和编译器。它旨在提供一个基于 Python 的编程环境，以高效编写自定义 DNN 计算内核，并能够在现代 GPU 硬件上以最大吞吐量运行。

更多 Triton 中文文档可访问 →https://triton.hyper.ai/

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

连接给定的块。

参数：

• input (Tensor) - 第一个输入张量。
• other (Tensor) - 第二个输入张量。
• reorder - 编译器提示。如果为 true，则允许编译器在连接输入时重新排序元素。仅在顺序不重要时使用（例如，结果仅用于缩减操作）。当前的 cat 实现仅支持 can_reorder=True。