diff --git a/docs/src/python/nn/layers.rst b/docs/src/python/nn/layers.rst index 4eb14b088..89afa8b4a 100644 --- a/docs/src/python/nn/layers.rst +++ b/docs/src/python/nn/layers.rst @@ -9,6 +9,8 @@ Layers :toctree: _autosummary :template: nn-module-template.rst + AdaptiveAvgPool2d + AdaptiveAvgPool3d ALiBi AvgPool1d AvgPool2d