Source code for hexagonal wave propagation and phase retrieval for hexagonal DOE design The implementation is based on PyTorch which supports efficient gradient propagation.
PyTorch, Matplotlib, Numpy
Example 0: To create the field in hexagonal coordinates and propagate the field.
Example 1: Phase retrieval solved by back-propagation and Gerchberg-Saxton.