triton.language.flip

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

Flips a tensor x along the dimension dim.

Parameters:

• x (Block) – the first input tensor

• dim (int) – the dimension to flip along

This function can also be called as a member function on tensor, as x.flip(...) instead of flip(x, ...).