mlx_identity function

int mlx_identity(
  1. Pointer<mlx_array> res,
  2. int n,
  3. mlx_dtype_ dtype,
  4. mlx_stream s,
)

Implementation

int mlx_identity(
  ffi.Pointer<mlx_array> res,
  int n,
  mlx_dtype_ dtype,
  mlx_stream s,
) => _mlx_identity(res, n, dtype.value, s);