ott.solvers.linear.implicit_differentiation.ImplicitDiff.gradient

ImplicitDiff.gradient(prob, f, g, lse_mode, gr)

Apply VJP to recover gradient in reverse mode differentiation.

Parameters:

• prob (LinearProblem) –

• f (Array) –

• g (Array) –

• lse_mode (bool) –

• gr (Tuple[Array, Array]) –

Return type:

LinearProblem