Very simple Pytorch implementation of gumbel-softmax trick for differentiable design of discrete optics
- such as the example of designing multi-level doe for hologram generation shown here.
Just run main.py
Optimize a size of
Final doe profile in 3D is: