flax
Adding "count_include_pad" argument to flax.linen.pooling.avg_pool
#2451
Merged
Merged