784d00f0c38fb5fd77aed0a6dffb93cc98a0c990,gpytorch/likelihoods/gaussian_likelihood.py,GaussianLikelihood,pyro_sample_y,#GaussianLikelihood#Any#Any#Any#Any#,87

Before Change

        y_dist = MultivariateNormal(y_mean, y_lazy_covar)
        if len(y_dist.batch_shape):
            y_dist = y_dist.__class__(
                y_dist.mean.contiguous().view(-1), BlockDiagLazyTensor(y_dist.lazy_covariance_matrix)
            )
            y_obs = y_obs.view_as(y_dist.mean)
        pyro.sample(name_prefix + "._training_labels", y_dist, obs=y_obs)

After Change

        // Samples will occur in the first batch dimension
        sample_shape = y_dist.shape()[:-y_obs.dim()]
        y_obs = y_obs.expand(y_dist.shape())
        with pyro.poutine.scale(scale=float(1. / sample_shape.numel())):
            pyro.sample(name_prefix + "._training_labels", y_dist, obs=y_obs)

Instances