dm-haiku
dm-haiku copied to clipboard
Adaptive pooling
Are there any plans to add adaptive pooling layer (where you only specify the desired output shape) to Haiku? If not would you consider a PR or rather keep layers simple?