feng-spinning/onn-simulation

A program using pytorch to simulate diffracive deep nerual network (D2NN), containing updates (relative to the original paper) of incoherent propagation, parallel computation, nonlinear activation, changing of propagating distance.

Python