ott.tools.gaussian_mixture.gaussian.Gaussian.from_random#

classmethod Gaussian.from_random(rng, n_dimensions, stdev_mean=0.1, stdev_cov=0.1, ridge=0, dtype=None)[source]#

Construct a random Gaussian.

Parameters

  • rng (PRNGKeyArray) – jax.random key

  • n_dimensions (int) – desired covariance dimensions

  • stdev_mean (float) – standard deviation of location and log eigenvalues (means for both are 0)

  • stdev_cov (float) – standard deviated of the covariance

  • ridge (Union[float, Array]) – Offset for means.

  • dtype (Optional[dtype]) – data type

Return type

Gaussian

Returns

A random Gaussian.