triton.language.flip¶

triton.language.flip(x, dim=None)¶

沿维度dim翻转张量x。

Parameters:

x (Block) – 第一个输入张量
dim (int) – 要翻转的维度（目前仅支持最后一个维度）

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