rtqichen / torchdiffeq

Differentiable ODE solvers with full GPU support and O(1)-memory backpropagation.
MIT License
5.61k stars 930 forks source link

Is func variable in odeint(func, y0, t) the derivative part of the ode? #237

Open erwusht opened 1 year ago

erwusht commented 1 year ago

In official documentation, the func is elaborate as "where func is any callable implementing the ordinary differential equation f(t, x)". Is it actually the derivative part of ode?