Function obj_func

python/examples/adjoint_optimization/multilayer_opt.py:209–220

Objective function for the optimization. Args: dft_ex: the discrete Fourier-transformed Ex fields as a 2D array of dimension (num_wavelengths, NX_DESIGN_GRID * NY_DESIGN_GRID). Returns: A tuple of the log of the integrals of Ex in the stack for each

(dft_ex: np.ndarray)

    ]

    def obj_func(dft_ex: np.ndarray) -> Tuple[float]:
        """Objective function for the optimization.

        Args:
          dft_ex: the discrete Fourier-transformed Ex fields as a 2D array of
            dimension (num_wavelengths, NX_DESIGN_GRID * NY_DESIGN_GRID).

        Returns:
          A tuple of the log of the integrals of Ex in the stack for each
          wavelength.
        """
        return npa.log(npa.sum(npa.absolute(dft_ex) ** 2, axis=1))

    opt = mpa.OptimizationProblem(
        simulation=sim,