jax
f5f35c5c - Adapted vmap out_axes check to sum_match

Commit

5 years ago

Adapted vmap out_axes check to sum_match

References

#2530 - Expand docstring for vmap with details about out_axes, and improve error checking

Author

gnecula

gnecula

Committer

gnecula

gnecula

Parents

Loading