jax
4fc73205 - Implement np.{diagonal,count_nonzero}.

Commit

7 years ago

Implement np.{diagonal,count_nonzero}. Fix shape error if a scalar is passed to reducers. Add test for scalar reductions.

References

Author

hawkinsp

Committer

hawkinsp

Parents