setu.linear_gaussian#

setu.linear_gaussian(theta, shift, cov, key)[source]#

Sample x from linear Gaussian likelihood: x ~ N(theta + shift, cov).

Parameters:

  • theta (Array) – Parameter values (batch, dim) or (dim,)

  • shift (Array) – Mean shift (dim,)

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

  • key (Array) – JAX random key

Return type:

Array

Returns:

Samples with same leading shape as theta