I am not able to differentiate code that involves a call to tfp.math.ode.DormandPrince() with complex arguments. I get a NotImplementedError. The error can be reproduced using the code below (based on #1372):
[/usr/local/lib/python3.8/dist-packages/tensorflow_probability/python/math/ode/base.py](https://localhost:8080/#) in error_if_complex(dtype)
249 def error_if_complex(dtype):
250 if dtype_util.is_complex(dtype):
--> 251 raise NotImplementedError('The adjoint sensitivity method does '
252 'not support complex dtypes.')
253
NotImplementedError: The adjoint sensitivity method does not support complex dtypes.
I'm wondering myself about this issue? I would be happy to take a stab at implementing this if experts think it is doable for a relative tensorflow novice
I am not able to differentiate code that involves a call to
tfp.math.ode.DormandPrince()
with complex arguments. I get aNotImplementedError
. The error can be reproduced using the code below (based on #1372):The relevant part of the error message is:
Will this capability be added soon?
The code works properly when the inputs are: