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 (currently only final dimension supported)