a7a12d157766b69cf4b1ddbb5fcdacfe485dc6fa,test/kernels/test_spectral_mixture_kernel.py,TestSpectralMixtureKernel,test_standard,#TestSpectralMixtureKernel#,10

Before Change

        kernel = SpectralMixtureKernel(num_mixtures=2, ard_num_dims=2)
        kernel.initialize(
            mixture_weights=torch.tensor([[4, 2]], dtype=torch.float),
            mixture_means=torch.tensor([[[[1, 2]], [[2, 1]]]], dtype=torch.float),
            mixture_scales=torch.tensor([[[[0.5, 0.25]], [[0.25, 0.5]]]], dtype=torch.float),
        )
        kernel.eval()

After Change

        a = torch.tensor([[0, 1], [2, 2], [2, 0]], dtype=torch.float)
        means = torch.tensor([[1, 2], [2, 1]], dtype=torch.float)
        scales = torch.tensor([[0.5, 0.25], [0.25, 0.5]], dtype=torch.float)
        scales = scales.unsqueeze(1)
        means = means.unsqueeze(1)
        weights = torch.tensor([4, 2], dtype=torch.float)
        kernel = SpectralMixtureKernel(num_mixtures=2, ard_num_dims=2)
        kernel.initialize(
            mixture_weights=weights,
            mixture_means=means,
            mixture_scales=scales,
        )
        kernel.eval()

        actual = torch.zeros(2, 3, 3, 2)
        for i in range(3):

Instances