Return real part of the gradient

2017-12-17 22:55:55 -08:00 · 2017-12-17 22:55:55 -08:00 · f312d73503
commit f312d73503
parent 39979edc44
1 changed files with 1 additions and 1 deletions
--- a/fdfd_tools/bloch.py
+++ b/fdfd_tools/bloch.py
@ -400,7 +400,7 @@ def eigsolve(num_modes: int,
            df_dy = scipy_iop @ (AzU - zU @ zTAzU)
        else:
            df_dy = (AzU - zU @ zTAzU)
-        return numpy.abs(f), numpy.sign(f) * df_dy.ravel()
+        return numpy.abs(f), numpy.sign(f) * numpy.real(df_dy).ravel()

    '''
    Use the conjugate gradient method and the approximate gradient calculation to