triton.language.view

triton.language.view(input, *shape, _semantic=None)

返回一个与 input 包含相同元素但形状不同的张量。元素的顺序可能无法保留。

参数:

• input (Block) – 输入张量。

• shape – 所需的形状。

shape 可以作为元组传递，也可以作为单独的参数传递。

# These are equivalent
view(x, (32, 32))
view(x, 32, 32)

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