diff --git a/docs/modules/layers.rst b/docs/modules/layers.rst index 618727c80..7e03e4d08 100644 --- a/docs/modules/layers.rst +++ b/docs/modules/layers.rst @@ -559,15 +559,15 @@ Pooling layer for any dimensions and any pooling functions. 1D Zero padding ^^^^^^^^^^^^^^^^^^^ -.. autofunction:: ZeroPad1d +.. autoclass:: ZeroPad1d 2D Zero padding ^^^^^^^^^^^^^^^^^^^ -.. autofunction:: ZeroPad2d +.. autoclass:: ZeroPad2d 3D Zero padding ^^^^^^^^^^^^^^^^^^^ -.. autofunction:: ZeroPad3d +.. autoclass:: ZeroPad3d 1D Max pooling ^^^^^^^^^^^^^^^^^^^