c8c7fc9f0c1778f79f758a968def34671b5cded4,gpytorch/lazy/interpolated_lazy_tensor.py,InterpolatedLazyTensor,zero_mean_mvn_samples,#InterpolatedLazyTensor#Any#,619

Before Change

        )

    def zero_mean_mvn_samples(self, n_samples):
        return left_interp(
            self.left_interp_indices, self.left_interp_values, self.base_lazy_tensor.zero_mean_mvn_samples(n_samples)
        )

    def __getitem__(self, index):
        index = list(index) if isinstance(index, tuple) else [index]
        ndimension = self.ndimension()

After Change

        )

    def zero_mean_mvn_samples(self, num_samples):
        if self.ndimension() == 3:
            res = left_interp(
                self.left_interp_indices, self.left_interp_values,
                self.base_lazy_tensor.zero_mean_mvn_samples(num_samples).permute(1, 2, 0).contiguous()
            )
            return res.permute(2, 0, 1).contiguous()
        else:
            res = left_interp(
                self.left_interp_indices, self.left_interp_values,
                self.base_lazy_tensor.zero_mean_mvn_samples(num_samples).permute(1, 0).contiguous()
            )
            return res.permute(1, 0).contiguous()

    def __getitem__(self, index):
        index = list(index) if isinstance(index, tuple) else [index]
        ndimension = self.ndimension()
        index += [slice(None, None, None)] * (ndimension - len(index))

Instances