mlx_random_truncated_normal function

int mlx_random_truncated_normal(
  1. Pointer<mlx_array> res,
  2. mlx_array lower,
  3. mlx_array upper,
  4. Pointer<Int> shape,
  5. int shape_num,
  6. mlx_dtype_ dtype,
  7. mlx_array key,
  8. mlx_stream s,
)

Implementation

int mlx_random_truncated_normal(
  ffi.Pointer<mlx_array> res,
  mlx_array lower,
  mlx_array upper,
  ffi.Pointer<ffi.Int> shape,
  int shape_num,
  mlx_dtype_ dtype,
  mlx_array key,
  mlx_stream s,
) => _mlx_random_truncated_normal(
  res,
  lower,
  upper,
  shape,
  shape_num,
  dtype.value,
  key,
  s,
);