Add Identity module name in __init__ (#39615)

* add_module_in_init_ * Update __init__.py * Update __init__.py
PaddlePaddle · Mar 29, 2022 · 869287f · 869287f
1 parent bea725b
commit 869287f
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/python/paddle/nn/__init__.py b/python/paddle/nn/__init__.py
@@ -306,4 +306,5 @@ def weight_norm(*args):
            'MaxUnPool2D',
            'MaxUnPool3D',
            'HingeEmbeddingLoss',
+           'Identity',
 ]