triton.language.view

triton.language.view(input, *shape)

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

Parameters:

• input (Block) – 输入张量。

• shape - 期望的形状。

shape 可以作为元组或单独参数传递：

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

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