gpytorch.kernels.keops

These kernels are compatible with the GPyTorch KeOps integration. For more information, see the KeOps tutorial.

Note

Only some standard kernels have KeOps impementations. If there is a kernel you want that’s missing, consider submitting a pull request!

RBFKernel

class gpytorch.kernels.keops.RBFKernel(ard_num_dims=None, batch_shape=None, active_dims=None, lengthscale_prior=None, lengthscale_constraint=None, eps=1e-06, **kwargs)[source]
Parameters:

  • ard_num_dims (int, optional) –

  • batch_shape (torch.Size, optional) –

  • active_dims ((int, ...), optional) –

  • lengthscale_prior (Prior, optional) –

  • lengthscale_constraint (Interval, optional) –

  • eps (float) –

MaternKernel

class gpytorch.kernels.keops.MaternKernel(nu=2.5, **kwargs)[source]
Parameters:

nu (float, optional) –

PeriodicKernel

class gpytorch.kernels.keops.PeriodicKernel(period_length_prior=None, period_length_constraint=None, **kwargs)[source]
Parameters:

  • period_length_prior (Prior, optional) –

  • period_length_constraint (Interval, optional) –