diff --git a/python/paddle/nn/__init__.py b/python/paddle/nn/__init__.py index c0820e140268b622f6c39aa86b955024076f343a..b83a900059bf491f33c231486b7a2efefc987bce 100644 --- a/python/paddle/nn/__init__.py +++ b/python/paddle/nn/__init__.py @@ -306,4 +306,5 @@ __all__ = [ #noqa 'MaxUnPool2D', 'MaxUnPool3D', 'HingeEmbeddingLoss', + 'Identity', ]