ConstantDiagonalKernelComputation

class gpjax.kernels.ConstantDiagonalKernelComputation

Bases: AbstractKernelComputation

Computation engine for constant diagonal kernels.

gram(kernel, x)

For a given kernel, compute Gram covariance operator of the kernel function on an input matrix of shape (N, D).

Parameters:

• kernel (K) – the kernel function.

• x (Float[jaxlib._jax.Array, 'N D'] | Float[ndarray, 'N D']) – the inputs to the kernel function of shape (N, D).

Returns:

The Gram covariance of the kernel function as a linear operator.

Return type:

AbstractLinearOperator