mlx_split function

  1. @Native<Int Function(Pointer<mlx_vector_array>, mlx_array, Int, Int, mlx_stream)>(ffi.Pointer<mlx_vector_array>, mlx_array, ffi.Int, ffi.Int, mlx_stream)>()
int mlx_split(
  1. Pointer<mlx_vector_array> res,
  2. mlx_array a,
  3. int num_splits,
  4. int axis,
  5. mlx_stream s,
)

Implementation

@ffi.Native<
  ffi.Int Function(
    ffi.Pointer<mlx_vector_array>,
    mlx_array,
    ffi.Int,
    ffi.Int,
    mlx_stream,
  )
>()
external int mlx_split(
  ffi.Pointer<mlx_vector_array> res,
  mlx_array a,
  int num_splits,
  int axis,
  mlx_stream s,
);