Ricky Chen
Results
2
issues of
Ricky Chen
Modify the wrapper (https://github.com/rtqichen/torchdiffeq/blob/master/torchdiffeq/_impl/scipy_wrapper.py) to take in an event_fn and wrap around the event functionality in scipy.solve_ivp.
Example code: ``` autograd = require 'autograd' f = function(x) return torch.add(x, -2, x) end df = autograd(f, {withForward = false, partialGrad = true}) x = torch.randn(3) df(x, torch.ones(3)) ```...