8e3272c530b75a2cdf25971068cba12316400d80,pyro/ops/gaussian.py,AffineNormal,condition,#AffineNormal#Any#,261

Before Change

        prec_sqrt = self.matrix / self.scale.unsqueeze(-2)
        precision = prec_sqrt.matmul(prec_sqrt.transpose(-1, -2))
        delta = (value - self.loc) / self.scale
        info_vec = prec_sqrt.matmul(delta.unsqueeze(-1)).squeeze(-1)
        log_normalizer = (-0.5 * self.loc.size(-1) * math.log(2 * math.pi)
                          - 0.5 * delta.pow(2).sum(-1) - self.scale.log().sum(-1))
        return Gaussian(log_normalizer, info_vec, precision)

After Change

        prec_sqrt = self.matrix / self.scale.unsqueeze(-2)
        precision = matmul(prec_sqrt, prec_sqrt.transpose(-1, -2))
        delta = (value - self.loc) / self.scale
        info_vec = matvecmul(prec_sqrt, delta)
        log_normalizer = (-0.5 * self.loc.size(-1) * math.log(2 * math.pi)
                          - 0.5 * delta.pow(2).sum(-1) - self.scale.log().sum(-1))
        return Gaussian(log_normalizer, info_vec, precision)

Instances