mlx_softmax function

  1. @Native<Int Function(Pointer<mlx_array>, mlx_array, Bool, mlx_stream)>(ffi.Pointer<mlx_array>, mlx_array, ffi.Bool, mlx_stream)>()
int mlx_softmax(
  1. Pointer<mlx_array> res,
  2. mlx_array a,
  3. bool precise,
  4. mlx_stream s,
)

Implementation

@ffi.Native<
  ffi.Int Function(ffi.Pointer<mlx_array>, mlx_array, ffi.Bool, mlx_stream)
>()
external int mlx_softmax(
  ffi.Pointer<mlx_array> res,
  mlx_array a,
  bool precise,
  mlx_stream s,
);