flax
c73a4b79
- Add count_include_pad argument to flax.linen.pooling.avg_pool / Fix div_shape for it raises error when there's no batch dimension
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
Commit
3 years ago
Add count_include_pad argument to flax.linen.pooling.avg_pool / Fix div_shape for it raises error when there's no batch dimension
References
#2451 - Adding "count_include_pad" argument to flax.linen.pooling.avg_pool
Author
dslisleedh
Parents
8af16050
Loading