Skip to contents

Sample from a normal distribution with mean \(\mu\) and standard deviation \(\sigma\).

Usage

nv_rnorm(shape, initial_state, dtype = "f32", mu = 0, sigma = 1)

Arguments

shape

(integer())
Shape.

initial_state

([ui64[2]][tensorish])
RNG state.

dtype

(character(1) | stablehlo::TensorDataType)
Data type.

mu

(numeric(1))
Expected value.

sigma

(numeric(1))
Standard deviation.

Value

(list() of tensorish)
List of two tensors: the new RNG state and the generated random numbers.

Covariance

To implement a covariance structure use cholesky decomposition.