ptyrad.core

PyTorch implementaiton of ptychography models, losses, and constraints

Modules

constraints	Physical constraints that directly modify optimizable tensors with specified intervals of iterations
forward	Physical forwad model that generates diffraction patterns from mixed-state probe/object in a fully vectorized way
functional	PyTorch implementation of core functionals like image shifts, blurring, DCT, etc.
losses	Loss functions and soft regularizations calculated using forward simulations against experimental measurements
models	Pytorch models for ptychography geometry, object, probe, scan patterns, etc.