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.4k stars 130 forks source link

update hybrid.py #74

Closed qpwodlsqp closed 3 years ago

qpwodlsqp commented 3 years ago

changed Y to have same dtypes and devices with input tensors

Zymrael commented 3 years ago

LGTM, thanks for the PR