bartorch.circshift

bartorch.circshift()

Cyclic shift along axes: element i moves to (i + shift) % n.

The convention of torch.roll().

Parameters:

• input (torch.Tensor) – Array to shift.

• shifts (int or tuple of int) – Shift in elements, one per entry of axes; may be negative.

• axes (int or tuple of int) – Axes to shift along.

Returns:

The shape of input.

Return type:

torch.Tensor