flax
0f9acdc8 - Fix DenseGeneral bug:

Commit
4 years ago
Fix DenseGeneral bug: - Correctly expand bias to account for batch & non-contracting dims. - Sort axes for consistent output under permutations
Author
Committer
Parents
Loading