flax
d4fa16d3 - Add test_avg_pool_no_batch under test.linen.linen_test.PoolTest

Commit

3 years ago

Add test_avg_pool_no_batch under test.linen.linen_test.PoolTest

References

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

Author

dslisleedh

dslisleedh

Parents

Loading