chromatix-team / chromatix

Differentiable wave optics using JAX! Documentation can be found at https://chromatix.readthedocs.io
MIT License
77 stars 13 forks source link

Quick fix of the vector field propagation #45

Closed nicolexi closed 1 year ago

nicolexi commented 1 year ago

remove line u = center_pad(field.u, [0, N_pad // 2, N_pad // 2, 0])