Commit b4282bfe authored by Reimar Leike's avatar Reimar Leike
Browse files

fix merge conflict

parents f2e283bf b2c5e03d
Pipeline #70291 passed with stages
in 15 minutes and 30 seconds
......@@ -128,11 +128,9 @@ class VariableCovarianceGaussianEnergy(EnergyOperator):
def apply(self, x):
res0 = x[self._r].vdot(x[self._r]*x[self._icov]).absolute()
res0 = x[self._r].vdot(x[self._r]*x[self._icov]).real
res1 = x[self._icov].log().sum()
res = 0.5*(res0-res1)
if (not isinstance(x, Linearization)) or (not x.want_metric):
return res
mf = {self._r: x.val[self._icov], self._icov: .5*x.val[self._icov]**(-2)}
metric = makeOp(MultiField.from_dict(mf))
return res.add_metric(SandwichOperator.make(x.jac, metric))
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