triton.language.view

triton.language.view(input, *shape)

返回一个张量，其元素与 input 相同，但形状不同。元素的顺序可能不被保留。

参数:

• input (Block) – 输入张量。

• shape – 期望的形状。

shape 可以作为元组或作为单独的参数传入

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

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