8e3272c530b75a2cdf25971068cba12316400d80,pyro/ops/gaussian.py,Gaussian,rsample,#Gaussian#Any#,136

Before Change

        loc = self.info_vec.unsqueeze(-1).cholesky_solve(P_chol).squeeze(-1)
        shape = sample_shape + self.batch_shape + (self.dim(), 1)
        noise = torch.randn(shape, dtype=loc.dtype, device=loc.device)
        noise = noise.triangular_solve(P_chol, upper=False, transpose=True).solution.squeeze(-1)
        return loc + noise

    def condition(self, value):

After Change

        loc = self.info_vec.unsqueeze(-1).cholesky_solve(P_chol).squeeze(-1)
        shape = sample_shape + self.batch_shape + (self.dim(), 1)
        noise = torch.randn(shape, dtype=loc.dtype, device=loc.device)
        noise = triangular_solve(noise, P_chol, upper=False, transpose=True).squeeze(-1)
        return loc + noise

    def condition(self, value):

Instances