The best way I found to implement the derivative is to use the grad-functionality of jax: https://jax.readthedocs.io/en/latest/notebooks/autodiff_cookbook.html
Good luck:)
The technical post webpages of this site follow the CC BY-SA 4.0 protocol. If you need to reprint, please indicate the site URL or the original address.Any question please contact:yoyou2525@163.com.