45d63060d155077f97e3cb42cf7551ec03af987c,gpytorch/kernels/grid_interpolation_kernel.py,GridInterpolationKernel,initialize_interpolation_grid,#GridInterpolationKernel#Any#Any#,14

Before Change

        self.grid_size = grid_size
        self.grid_bounds = grid_bounds
        self.grid = Variable(torch.linspace(grid_bounds[0] - grid_diff,
                                            grid_bounds[1] + grid_diff,
                                            grid_size))

        return self

After Change

        super(GridInterpolationKernel, self).initialize_interpolation_grid(grid_size, grid_bounds)
        self.grid_size = grid_size
        self.grid_bounds = grid_bounds
        self.grid = torch.zeros(len(grid_bounds), grid_size)
        for i in range(len(grid_bounds)):
            grid_diff = (grid_bounds[i][1] - grid_bounds[i][0]) / (grid_size - 2)
            self.grid[i] = torch.linspace(grid_bounds[i][0] - grid_diff,
                                          grid_bounds[i][1] + grid_diff,
                                          grid_size)
        self.grid = Variable(self.grid)
        return self

    def forward(self, x1, x2, **kwargs):
        n, d = x1.size()

Instances