dirac

deepinv.utils.dirac(shape, device='cpu')

Returns a torch.Tensor with a Dirac delta in 2D at the center.

Parameters:

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

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