71c4ebaf3ecffd66f43a575a010808dae4266b8c,gpytorch/random_variables/gaussian_random_variable.py,GaussianRandomVariable,sample,#GaussianRandomVariable#Any#,41

Before Change

        return self._mean, self._covar

    def sample(self, n_samples):
        raise NotImplementedError

    def var(self):
        return self._covar.diag()

After Change

        return self._mean, self._covar

    def sample(self, n_samples):
        base_samples = Variable(self._mean.data.new(self._covar.chol_approx_size(), n_samples).normal_())
        samples = self._covar.chol_matmul(base_samples)
        samples = samples + self._mean.unsqueeze(-1)
        return samples

    def var(self):
        return self._covar.diag()

Instances