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)

References

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

Author

hawkinsp

hawkinsp

Parents

Loading