f47e301e50018af3365e3717e3ab1d8e07d297b5,gpytorch/kernels/grid_kernel.py,GridKernel,__init__,#GridKernel#Any#Any#Any#,9

Before Change

    def __init__(self, base_kernel_module, inducing_points, grid):
        super(GridKernel, self).__init__()
        self.base_kernel_module = base_kernel_module
        if inducing_points.ndimension() != 2:
            raise RuntimeError("Inducing points should be 2 dimensional")
        self.register_buffer("inducing_points", inducing_points.unsqueeze(0))
        self.register_buffer("grid", grid)

    def train(self, mode=True):

After Change

class GridKernel(Kernel):
    def __init__(self, base_kernel_module, inducing_points, grid, active_dims=None):
        super(GridKernel, self).__init__(active_dims=active_dims)
        self.base_kernel_module = base_kernel_module
        if inducing_points.ndimension() != 2:
            raise RuntimeError("Inducing points should be 2 dimensional")
        self.register_buffer("inducing_points", inducing_points.unsqueeze(0))
        self.register_buffer("grid", grid)

    def train(self, mode=True):

Instances