Skip to content

fixing avg_pool wrong shape issue#824

Open
Vasist10 wants to merge 1 commit intogoogle-deepmind:mainfrom
Vasist10:fix/avg-pool-padding-shape
Open

fixing avg_pool wrong shape issue#824
Vasist10 wants to merge 1 commit intogoogle-deepmind:mainfrom
Vasist10:fix/avg-pool-padding-shape

Conversation

@Vasist10
Copy link
Copy Markdown

@Vasist10 Vasist10 commented Jun 1, 2025

issue link : https://github.com/google-deepmind/dm-haiku/issues/822

This PR improves avg_pool by:

  • Skipping pooling over channel_axis
  • fixing the shape of the output
  • Handling negative axis

@Vasist10
Copy link
Copy Markdown
Author

Hi, just following up to see if there’s any feedback or changes needed on this PR. Happy to make updates if needed. Thanks!

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant