search

DiffEqML / torchdyn

A PyTorch library entirely dedicated to neural differential equations, implicit models and related numerical methods
https://torchdyn.org
Apache License 2.0
1.36k stars 125 forks source link

A new model? #81

Closed jbgao closed 3 years ago

jbgao commented 3 years ago

Describe the solution you'd like

If my vector field function is in the form of f(z(t), x(t)), where z the solution of ODE where x(t) is a known control input, can we have a model in TorchDyn for this type ODE. Or this has been considered? Any hints or suggestions?

J.

Zymrael commented 3 years ago

This can be easily done by concatenating z, x and then using them inside the ODE function. We have examples of this in our optimal control tutorials.