diff --git a/imperative/python/megengine/functional/nn.py b/imperative/python/megengine/functional/nn.py
index d2acbc1b..61e5ba2d 100644
--- a/imperative/python/megengine/functional/nn.py
+++ b/imperative/python/megengine/functional/nn.py
@@ -42,7 +42,7 @@ from ..utils.deprecation import deprecated_func
 from .debug_param import get_execution_strategy
 from .distributed import all_reduce_sum
 from .elemwise import _elwise, exp, log, log1p, maximum, minimum
-from .math import max, sum
+from .math import max, normalize, sum
 from .metric import topk_accuracy
 from .tensor import broadcast_to, concat, expand_dims, ones, squeeze, zeros
 
@@ -73,6 +73,7 @@ __all__ = [
     "logsumexp",
     "logsoftmax",
     "max_pool2d",
+    "normalize",
     "one_hot",
     "prelu",
     "pad",