JaxBackend.roll

JaxBackend.roll(a, shift, axis, **kwargs)

Roll array elements along a specified axis.

Parameters:

aBackendArray

Input data.

shiftint or tuple of ints, optional

Shift along each axis.

Returns:

BackendArray

Array with elements rolled.