gradoptics.optics.psf
- class gradoptics.optics.psf.PSF(kernel)
Bases:
torch.nn.modules.module.ModuleModels a point spread function.
- __init__(kernel)
- Parameters
kernel – Kernel that characterizes the point spread function (
torch.tensor)
- forward(x)
Performs a convolution between the input data and the kernel of the point spread function
- Parameters
x – Data to convolve with the PSF (
torch.tensor)- Returns
The convolved input (
torch.tensor)
- training: bool