jax
a9da06ce - Fix shape error when taking JVP of reduce-prod over size 0 axis. (#3729)

Commit
5 years ago
Fix shape error when taking JVP of reduce-prod over size 0 axis. (#3729)
Author
Parents
Loading