Commit c1fdf306 authored by Philipp Arras's avatar Philipp Arras
Browse files

Support complex data

parent 59a72c4b
...@@ -159,7 +159,7 @@ class VariableCovarianceGaussianEnergy(EnergyOperator): ...@@ -159,7 +159,7 @@ class VariableCovarianceGaussianEnergy(EnergyOperator):
def apply(self, x): def apply(self, x):
self._check_input(x) self._check_input(x)
res = 0.5*(x[self._r].vdot(x[self._r]*x[self._icov]).real - x[self._icov].ptw("log").sum()) res = 0.5*(x[self._r].vdot(x[self._r]*x[self._icov].real).real - x[self._icov].ptw("log").sum())
if not x.want_metric: if not x.want_metric:
return res return res
mf = {self._r: x.val[self._icov], self._icov: .5*x.val[self._icov]**(-2)} mf = {self._r: x.val[self._icov], self._icov: .5*x.val[self._icov]**(-2)}
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment