def ifftshift()

in activemri/experimental/cvpr19_models/models/fft_utils.py [0:0]

• 9 lines of code
• 5 McCabe index (conditional complexity)

def ifftshift(x, dim=None):
    if dim is None:
        dim = tuple(range(x.dim()))
        shift = [(dim + 1) // 2 for dim in x.shape]
    elif isinstance(dim, int):
        shift = (x.shape[dim] + 1) // 2
    else:
        shift = [(x.shape[i] + 1) // 2 for i in dim]
    return roll(x, shift, dim)