def exact_posterior_mean(self, test_mean, alpha): if isinstance(self.var, LazyVariable): return self.var.matmul(alpha) + test_mean return torch.addmv(test_mean, self.var, alpha)