Ptychography#

class deepinv.physics.Ptychography(in_shape=None, probe=None, shifts=None, device='cpu', **kwargs)[source]#

Ptychography forward operator. Corresponding to the operator

\[\forw{x} = \left| Bx \right|^2\]

where \(B\) is the linear forward operator defined by a deepinv.physics.PtychographyLinearOperator object.

Parameters:

in_shape (tuple) – Shape of the input image.
probe (None,torch.Tensor) – A tensor of shape img_size representing the probe function. If None, a disk probe is generated with deepinv.physics.phase_retrieval.build_probe function.
shifts (None,array_like) – A 2D array of shape (n_img, 2) corresponding to the shifts for the probe. If None, shifts are generated with deepinv.physics.phase_retrieval.generate_shifts function.
device (torch.device, str) – Device “cpu” or “gpu”.

Examples using `Ptychography`:#