diff --git a/python/paddle/nn/__init__.py b/python/paddle/nn/__init__.py index 1abe74e9783dc4a6eb1767f547a75a78815e68c0..3afd2b56569a497b753b49fe7c1b95a09873e5be 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 b5d6d7834f95924fbfe067c1c3d760e307b1f8f4..cc49db9b2056fc66be3c9d6cd53ed77b25c635d9 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] """