10cb719041b531cb25e485f928ccb545a475a71a,gpytorch/kernels/rbf_kernel.py,RBFKernel,forward,#RBFKernel#Any#Any#,104

Before Change

    def forward(self, x1, x2):
        lengthscales = self.log_lengthscale.exp().mul(math.sqrt(2)).clamp(self.eps, 1e5)
        diff = (x1.unsqueeze(2) - x2.unsqueeze(1)).div_(lengthscales.unsqueeze(1))
        return diff.pow_(2).sum(-1).mul_(-1).exp_()

After Change

    def forward(self, x1, x2):
        x1_, x2_ = self._create_input_grid(x1, x2)
        x1_ = x1_.div(self.lengthscale)
        x2_ = x2_.div(self.lengthscale)

        diff = (x1_ - x2_).norm(2, dim=-1)
        return diff.pow(2).div_(-2).exp_()

Instances