b821df469dbf44ab1f1f3564c454eb2787cd412b,test/kernels/test_rbf_kernel.py,TestRBFKernel,test_ard_batch,#TestRBFKernel#,48

Before Change

        // diag
        res = kernel(a, b).diag()
        actual = torch.cat([actual[i].diag().unsqueeze(0) for i in range(actual.size(0))])
        self.assertLess(torch.norm(res - actual), 1e-5)

        // batch_dims
        double_batch_a = scaled_a.unsqueeze(0).transpose(0, -1)

After Change

        b = torch.tensor([[[1, 3, 1]], [[2, -1, 0]]], dtype=torch.float).repeat(1, 2, 1)
        lengthscales = torch.tensor([[[1, 2, 1]]], dtype=torch.float)

        kernel = RBFKernel(batch_shape=torch.Size([2]), ard_num_dims=3)
        kernel.initialize(lengthscale=lengthscales)
        kernel.eval()

Instances