triton.language.flip

triton.language.flip = JITFunction(triton.triton_patch.language.standard:flip)

Flips a tensor x along the dimension dim.

Parameters:

• x (Block) – the first input tensor

• dim (int) – the dimension to flip along (currently only final dimension supported)

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