softmin function is included

Author: CarlaRodriguez96

xlumina/toolbox.py

@@ -13,6 +13,7 @@
     - space
     - wrap_phase
     - is_conserving_energy
+    - softmin
     - delta_kronecker
     - build_LCD_cell
     - draw_sSLM
@@ -78,6 +79,14 @@ def is_conserving_energy(light_source, propagated_light):
 
     return I_propagated / I_source
 
+@jit
+def softmin(args, beta=90):
+    """
+    Differentiable version for min() function.
+    """
+    return - nn.logsumexp(-beta * args) / beta
+
+
 def delta_kronecker(a, b):
     """
     Computes the Kronecker delta.