bartorch.fftshift

bartorch.fftshift()

Move index zero to index n // 2 along axes, or back with inverse.

Parameters:

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

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

• inverse (bool, default=False) – Shift index n // 2 back to index zero; differs from the forward shift for odd sizes.

Returns:

The shape of input.

Return type:

torch.Tensor