d373facb5c282092776f4bcef09ec925d3a651e3,gpytorch/kernels/kernel.py,Kernel,__call__,#Kernel#Any#Any#,37

Before Change

        if self.active_dims is not None:
            x1 = x1_[:, self.active_dims]
            if x2_ is not None:
                x2 = x2_[:, self.active_dims]
        else:
            x1 = x1_
            x2 = x2_

After Change

    def __call__(self, x1_, x2_=None, **params):
        if self.active_dims is not None:
            x1 = x1_.index_select(-1, self.active_dims)
            if x2_ is not None:
                x2 = x2_.index_select(-1, self.active_dims)
        else:
            x1 = x1_

Instances