ott.solvers.linear.implicit_differentiation.ImplicitDiff.gradient# ImplicitDiff.gradient(prob, f, g, lse_mode, gr)[source]# 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
