neural_tangents.predict.Gaussian

class neural_tangents.predict.Gaussian(mean: jax.numpy.ndarray, covariance: jax.numpy.ndarray)[source]

A (mean, covariance) convenience namedtuple.

mean

Mean of shape equal to the shape of the function outputs.

Type

jax.numpy.ndarray

covariance

Covariance of shape equal to the shape of the respective NTK/NNGP kernel.

Type

jax.numpy.ndarray

__init__()

Methods

__init__()

count(value, /)

Return number of occurrences of value.

index(value[, start, stop])

Return first index of value.

Attributes

covariance

Alias for field number 1

mean

Alias for field number 0