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.
- covariance
Covariance of shape equal to the shape of the respective NTK/NNGP kernel.
- __init__()
Methods
__init__
()count
(value, /)Return number of occurrences of value.
index
(value[, start, stop])Return first index of value.
Attributes
Alias for field number 1
Alias for field number 0