0f8802fba8f55cd66300017fdf9b7189183ff9d5,test/kernels/test_scale_kernel.py,TestScaleKernel,test_ard_batch,#TestScaleKernel#,44

Before Change

        self.assertLess(torch.norm(res - actual), 1e-5)

        // batch_dims
        double_batch_a = scaled_a.unsqueeze(0).transpose(0, -1)
        double_batch_b = scaled_b.unsqueeze().transpose(0, -1)
        actual = double_batch_a.transpose(-1, -2).unsqueeze(-1) - double_batch_b.transpose(-1, -2).unsqueeze(-2)
        actual = actual.pow(2).mul_(-0.5).exp().view(3, 2, 2, 2)
        actual[:, 1, :, :].mul_(2)
        res = kernel(a, b, batch_dims=(0, 2)).evaluate()
        self.assertLess(torch.norm(res - actual), 1e-5)

After Change

        base_kernel = RBFKernel(batch_shape=torch.Size([2]), ard_num_dims=3)
        base_kernel.initialize(lengthscale=lengthscales)
        kernel = ScaleKernel(base_kernel, batch_shape=torch.Size([2]))
        kernel.initialize(outputscale=torch.tensor([1, 2], dtype=torch.float))
        kernel.eval()

Instances