setu.sample_prior#

setu.sample_prior(mean, cov, n_samples, key)[source]#

Sample theta from multivariate normal prior.

Parameters:

  • mean (Array) – Prior mean (dim,)

  • cov (Array) – Prior covariance (dim, dim)

  • n_samples (int) – Number of samples

  • key (Array) – JAX random key

Return type:

Array

Returns:

Samples (n_samples, dim)