triton.language.extra.cann.extension.flip

triton.language.extra.cann.extension.flip(ptr, dim=-1, _semantic=None, _generator=None)

Flips a tensor along the specified dimension.

Reverses the order of elements along the given axis. This is an Ascend-specific implementation that supports both SIMD and non-SIMD execution modes.

参数:

• ptr (tensor) -- The input tensor to flip.

• dim (int) -- The dimension along which to flip. Defaults to -1 (last dimension).