e503f50cb38fb1864fb6d66ba927b0852159428d,gpytorch/kernels/spectral_mixture_kernel.py,SpectralMixtureKernel,forward,#SpectralMixtureKernel#Any#Any#,18

Before Change

        sq_distance.mul_(2 * math.pi ** 2)  // sq_distance = -2*pi^2*(x-z)^2

        res = torch.zeros(n, m)
        for weight, mean, scale in zip(mixture_weights, mixture_means, mixture_scales):
            weight = weight.expand(n, m)
            mean = mean.expand(n, m)
            scale = scale.expand(n, m)

After Change

        sq_distance.mul_(2 * math.pi ** 2)  // sq_distance = -2*pi^2*(x-z)^2

        res = x1.data.new(n, m).zero_()
        for weight, mean, scale in zip(mixture_weights, mixture_means, mixture_scales):
            weight = weight.expand(n, m)
            mean = mean.expand(n, m)
            scale = scale.expand(n, m)

Instances