Skip to main content

mlx_tensordot

Function mlx_tensordot 

Source
pub unsafe extern "C" fn mlx_tensordot(
    res: *mut mlx_array,
    a: mlx_array,
    b: mlx_array,
    axes_a: *const c_int,
    axes_a_num: usize,
    axes_b: *const c_int,
    axes_b_num: usize,
    s: mlx_stream,
) -> c_int