Sample from a normal distribution on mlx arrays

Sample from a normal distribution on mlx arrays

Usage

mlx_rand_normal(
  dim,
  mean = 0,
  sd = 1,
  dtype = c("float32", "float64"),
  device = mlx_default_device()
)

Arguments

dim

Integer vector specifying array dimensions (shape).

mean

Mean of the normal distribution.

sd

Standard deviation of the normal distribution.

dtype

Desired MLX dtype ("float32" or "float64").

device

Execution target: supply "gpu", "cpu", or an mlx_stream created via mlx_new_stream(). Defaults to the current mlx_default_device() unless noted otherwise (helpers that act on an existing array typically reuse that array's device or stream).

Value

An mlx array with normally distributed entries.

See also

mlx.core.random.normal

Examples

weights <- mlx_rand_normal(c(3, 3), mean = 0, sd = 0.1)