84fccd898c45c08279fb5c109e6e234f3a47588a,test/kernels/test_rbf_kernel.py,TestRBFKernel,test_computes_radial_basis_function_gradient,#TestRBFKernel#,63

Before Change

        kernel = RBFKernel().initialize(log_lengthscale=math.log(lengthscale))
        kernel.eval()
        param = Variable(
            torch.Tensor(3, 3).fill_(math.log(lengthscale)),
            requires_grad=True,
        )
        diffs = Variable(a.expand(3, 3) - b.expand(3, 3).transpose(0, 1))
        actual_output = (-(diffs ** 2) / (param.exp())).exp()
        actual_output.backward(torch.eye(3))
        actual_param_grad = param.grad.data.sum()

        output = kernel(Variable(a), Variable(b))
        output.backward(gradient=torch.eye(3))

After Change

        kernel = RBFKernel().initialize(log_lengthscale=math.log(lengthscale))
        kernel.eval()

        param = math.log(lengthscale) * torch.ones(3, 3)
        param.requires_grad_()
        diffs = a.expand(3, 3) - b.expand(3, 3).transpose(0, 1)
        actual_output = (-0.5 * (diffs / param.exp()) ** 2).exp()
        actual_output.backward(gradient=torch.eye(3))
        actual_param_grad = param.grad.sum()

        output = kernel(a, b)
        output.backward(gradient=torch.eye(3))

Instances