dirac_comb

deepinv.utils.dirac_comb(shape, step, device='cpu')

Returns a torch.Tensor with a Dirac comb in 2D (impulse train) at the given step.

Parameters:

• shape (tuple) – shape of the output tensor.

• step (int) – step of the Dirac comb.

• device (str) – device of the output tensor.

Examples using dirac_comb:

Tour of blur operators

Tour of blur operators