From 890bd6266c1ba638ded7487e189fcf658e0579a1 Mon Sep 17 00:00:00 2001
From: xiaoting <31891223+tink2123@users.noreply.github.com>
Date: Tue, 7 Dec 2021 17:32:49 +0800
Subject: [PATCH] add maxunpool2d in __all__ (#37698)

* add maxunpool2d in __all__

* fix MaxUnPool2D example
---
 python/paddle/nn/__init__.py      | 3 ++-
 python/paddle/nn/layer/pooling.py | 2 +-
 2 files changed, 3 insertions(+), 2 deletions(-)

diff --git a/python/paddle/nn/__init__.py b/python/paddle/nn/__init__.py
index 1abe74e978..3afd2b5656 100644
--- a/python/paddle/nn/__init__.py
+++ b/python/paddle/nn/__init__.py
@@ -295,5 +295,6 @@ __all__ = [     #noqa
            'ELU',
            'ReLU6',
            'LayerDict',
-           'ZeroPad2D'
+           'ZeroPad2D',
+           'MaxUnPool2D',
 ]
diff --git a/python/paddle/nn/layer/pooling.py b/python/paddle/nn/layer/pooling.py
index b5d6d7834f..cc49db9b20 100755
--- 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]
 
     """
-- 
GitLab