autopdex.solution_structures.precompute_function_and_derivatives

autopdex.solution_structures.precompute_function_and_derivatives(fun, x_int, num_diff=1)

Precomputes a function and its derivatives up to a specified order at given integration points.

Parameters:

• fun (callable) – The function to precompute.

• x_int (jnp.ndarray) – Integration points.

• num_diff (int, optional) – Number of derivatives to compute (0, 1, or 2). Default is 1.

Returns:

Precomputed function values and derivatives.

• If num_diff == 0: (fun_i,)

• If num_diff == 1: (fun_i, dfun_i)

• If num_diff == 2: (fun_i, dfun_i, ddfun_i)

Return type:

tuple