build_probe#
- class deepinv.physics.phase_retrieval.build_probe(img_size, type='disk', probe_radius=10, device='cpu')[source]#
Bases:
Builds a probe based on the specified type and radius.
- Parameters:
img_size (tuple) – Shape of the input image.
type (str) – Type of probe shape, e.g., “disk”.
probe_radius (int) – Radius of the probe shape.
device (torch.device) – Device “cpu” or “gpu”.
- Returns:
Tensor representing the constructed probe.
