a7a12d157766b69cf4b1ddbb5fcdacfe485dc6fa,gpytorch/kernels/scale_kernel.py,ScaleKernel,forward,#ScaleKernel#Any#Any#Any#Any#,80

Before Change

    def forward(self, x1, x2, batch_dims=None, diag=False, **params):
        outputscales = self.outputscale
        if batch_dims == (0, 2) and outputscales.numel() > 1:
            outputscales = outputscales.unsqueeze(1).repeat(1, x1.size(-1)).view(-1)

        orig_output = self.base_kernel.forward(x1, x2, diag=diag, batch_dims=batch_dims, **params)
        outputscales = outputscales.view(-1, *([1] * (orig_output.dim() - 1)))

After Change

            outputscales = outputscales.unsqueeze(0).repeat(x1.size(-1), *([1] * outputscales.dim()))

        orig_output = self.base_kernel.forward(x1, x2, diag=diag, batch_dims=batch_dims, **params)
        outputscales = outputscales.view(*outputscales.shape, *([1] * (orig_output.dim() - outputscales.dim())))

        if diag:
            return delazify(orig_output) * outputscales

Instances