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, ...)。