pub unsafe extern "C" fn ggml_soft_max_ext(
    ctx: *mut ggml_context,
    a: *mut ggml_tensor,
    mask: *mut ggml_tensor,
    scale: f32
) -> *mut ggml_tensor