White
White
White(active_dims=None, variance=1.0, n_dims=None, compute_engine=ConstantDiagonalKernelComputation())
Bases: StationaryKernel
The White noise kernel.
Computes the covariance for pairs of inputs \((x, y)\) with variance \(\sigma^2\): $$ k(x, y) = \sigma^2 \delta(x-y) $$
Parameters:
-
active_dims
(Union[list[int], slice, None]
, default:None
) βThe indices of the input dimensions that the kernel operates on.
-
variance
(Union[ScalarFloat, Variable[ScalarArray]]
, default:1.0
) βthe variance of the kernel Ο.
-
n_dims
(Union[int, None]
, default:None
) βThe number of input dimensions.
-
compute_engine
(AbstractKernelComputation
, default:ConstantDiagonalKernelComputation()
) βThe computation engine that the kernel uses to compute the covariance matrix
spectral_density
property
The spectral density of the kernel.
Returns:
-
Distribution
βCallable[[Float[Array, "D"]], Float[Array, "D"]]: The spectral density function.
cross_covariance
Compute the cross-covariance matrix of the kernel.
Parameters:
-
x
(Num[Array, 'N D']
) βthe first input matrix of shape
(N, D)
. -
y
(Num[Array, 'M D']
) βthe second input matrix of shape
(M, D)
.
Returns:
-
Float[Array, 'N M']
βThe cross-covariance matrix of the kernel of shape
(N, M)
.
gram
Compute the gram matrix of the kernel.
Parameters:
-
x
(Num[Array, 'N D']
) βthe input matrix of shape
(N, D)
.
Returns:
-
LinearOperator
βThe gram matrix of the kernel of shape
(N, N)
.
diagonal
Compute the diagonal of the gram matrix of the kernel.
Parameters:
-
x
(Num[Array, 'N D']
) βthe input matrix of shape
(N, D)
.
Returns:
-
Float[Array, ' N']
βThe diagonal of the gram matrix of the kernel of shape
(N,)
.
slice_input
Slice out the relevant columns of the input matrix.
Select the relevant columns of the supplied matrix to be used within the kernel's evaluation.
Parameters:
-
x
(Float[Array, '... D']
) βthe matrix or vector that is to be sliced.
Returns:
-
Float[Array, '... Q']
βThe sliced form of the input matrix.
__add__
Add two kernels together. Args: other (AbstractKernel): The kernel to be added to the current kernel.
Returns:
-
AbstractKernel
(AbstractKernel
) βA new kernel that is the sum of the two kernels.
__mul__
Multiply two kernels together.
Parameters:
-
other
(AbstractKernel
) βThe kernel to be multiplied with the current kernel.
Returns:
-
AbstractKernel
(AbstractKernel
) βA new kernel that is the product of the two kernels.