mlx_trace function

int mlx_trace(
  1. Pointer<mlx_array> res,
  2. mlx_array a,
  3. int offset,
  4. int axis1,
  5. int axis2,
  6. mlx_dtype_ dtype,
  7. mlx_stream s,
)

Implementation

int mlx_trace(
  ffi.Pointer<mlx_array> res,
  mlx_array a,
  int offset,
  int axis1,
  int axis2,
  mlx_dtype_ dtype,
  mlx_stream s,
) => _mlx_trace(res, a, offset, axis1, axis2, dtype.value, s);