8df47901fbae1c93bcec58a9a658a6a3ad9f5d07,pymc3/variational/flows.py,ReferencePointFlow,logdet,#ReferencePointFlow#,303

Before Change

        har = h(a, r)
        dar = deriv(a, r)
        det = (1. + b*har)**(d-1) * (1. + b*har + b*dar*r)
        return tt.log(det)


class Radial(FlowFn):
    @staticmethod

After Change

        z = self.z0  // sxd
        h = self.h  // h(a, r)
        deriv = self.h.deriv  // h"(a, r)
        if self.is_local:
            z = z.swapaxes(0, 1)
        r = (z - z_ref).norm(2, axis=-1, keepdims=True)  // s
        har = h(a, r)
        dar = deriv(a, r)
        logdet = tt.log((1. + b*har)**(d-1) * (1. + b*har + b*dar*r))
        if self.is_local:
            logdet = logdet.sum(0)
        return logdet


class Radial(FlowFn):
    @staticmethod

Instances