【Triton 教程】triton_language.view

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

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

triton.language.view(input, *shape)

返回具有与输入相同元素但形状不同的张量，元素的顺序可能无法保持。

参数**：**

• input (Block) - 输入张量。
• shape - 所需的形状。

shape 可以作为 1 个元组或作为单独的参数被传递：

# These are equivalent
# 这些是等价的
view(x, (32, 32))
view(x, 32, 32)

这个函数也可作为 tensor 的成员函数调用，使用 x.view(...) 而不是 view(x, ...)。