Return real part of the gradient

This commit is contained in:
jan 2017-12-17 22:55:55 -08:00
parent 39979edc44
commit f312d73503

View File

@ -400,7 +400,7 @@ def eigsolve(num_modes: int,
df_dy = scipy_iop @ (AzU - zU @ zTAzU) df_dy = scipy_iop @ (AzU - zU @ zTAzU)
else: else:
df_dy = (AzU - zU @ zTAzU) 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 Use the conjugate gradient method and the approximate gradient calculation to