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.

Return type

LinearProblem

Parameters