add maxunpool2d in __all__ (#37698)

* add maxunpool2d in __all__ * fix MaxUnPool2D example

add maxunpool2d in all (#37698)
* add maxunpool2d in __all__ * fix MaxUnPool2D example
890bd626 · xiaoting · GitHub · ca6ff1f6 · 890bd626 · 890bd626
显示空白变更内容
内联并排

Showing with 3 addition and 2 deletion

python/paddle/nn/__init__.py python/paddle/nn/__init__.py +2 -1

python/paddle/nn/layer/pooling.py python/paddle/nn/layer/pooling.py +1 -1

未找到文件。
--- a/python/paddle/nn/__init__.py
+++ b/python/paddle/nn/__init__.py
@@ -295,5 +295,6 @@ __all__ = [     #noqa
           'ELU',
           'ReLU6',
           'LayerDict',
-           'ZeroPad2D'
+           'ZeroPad2D',
+           'MaxUnPool2D',
 ]
--- a/python/paddle/nn/layer/pooling.py
+++ b/python/paddle/nn/layer/pooling.py
@@ -1181,7 +1181,7 @@ class MaxUnPool2D(Layer):
        pool_out, indices = F.max_pool2d(data, kernel_size=2, stride=2, padding=0, return_mask=True)
        # pool_out shape: [1, 1, 3, 3],  indices shape: [1, 1, 3, 3]
        Unpool2D = paddle.nn.MaxUnPool2D(kernel_size=2, padding=0)
-        unpool_out = UnPool2D(pool_out, indices)
+        unpool_out = Unpool2D(pool_out, indices)
        # unpool_out shape: [1, 1, 6, 6]

    """