Martin Reinecke committed Aug 23, 2017 1 2 ``````from .energy import Energy from .memoization import memo `````` Martin Reinecke committed Aug 19, 2017 3 4 5 6 `````` class QuadraticEnergy(Energy): """The Energy for a quadratic form. `````` Martin Reinecke committed Aug 19, 2017 7 8 `````` The most important aspect of this energy is that its curvature must be position-independent. `````` Martin Reinecke committed Aug 19, 2017 9 10 `````` """ `````` Martin Reinecke committed Sep 29, 2017 11 `````` def __init__(self, position, A, b, _grad=None): `````` Martin Reinecke committed Aug 19, 2017 12 13 14 `````` super(QuadraticEnergy, self).__init__(position=position) self._A = A self._b = b `````` Martin Reinecke committed Sep 07, 2017 15 16 `````` if _grad is not None: self._Ax = _grad + self._b `````` Martin Reinecke committed Aug 20, 2017 17 18 `````` else: self._Ax = self._A(self.position) `````` Martin Reinecke committed Aug 19, 2017 19 20 `````` def at(self, position): `````` Martin Reinecke committed Sep 29, 2017 21 `````` return QuadraticEnergy(position=position, A=self._A, b=self._b) `````` Martin Reinecke committed Aug 19, 2017 22 `````` `````` Martin Reinecke committed Aug 20, 2017 23 `````` def at_with_grad(self, position, grad): `````` Martin Reinecke committed Sep 29, 2017 24 25 `````` return QuadraticEnergy(position=position, A=self._A, b=self._b, _grad=grad) `````` Martin Reinecke committed Aug 20, 2017 26 `````` `````` Martin Reinecke committed Aug 19, 2017 27 28 29 30 31 32 33 34 35 36 37 38 39 `````` @property @memo def value(self): return 0.5*self.position.vdot(self._Ax) - self._b.vdot(self.position) @property @memo def gradient(self): return self._Ax - self._b @property def curvature(self): return self._A``````