【Triton 教程】triton_language.tensor
💡
原文中文，约2800字，阅读约需7分钟。
📝
内容提要
Triton 是一种基于 Python 的并行编程语言和编译器，专为高效编写自定义 DNN 计算内核而设计，支持现代 GPU 硬件。其核心数据结构为 N 维数组 tensor，并提供多种操作函数。
🎯
关键要点
- Triton 是一种基于 Python 的并行编程语言和编译器，专为高效编写自定义 DNN 计算内核而设计。
- Triton 支持现代 GPU 硬件，以最大吞吐量运行。
- Triton 的核心数据结构是 N 维数组 tensor，提供多种操作函数。
- 在 Triton 程序中，tensor 是最基本的数据结构，许多函数对 tensors 进行操作并返回。
- tensor 类定义了多个方法，转发到 triton.language 中的自由函数。
- 构造函数和大多数方法不被用户代码直接调用。
- tensor 类支持多种数学和逻辑操作，如加法、乘法、最大值、最小值等。
- tensor 类还定义了转置等属性，便于操作 2D 张量。
➡️
