From 213f40437c402bc79c7e74fd352c9e56b5ff8f50 Mon Sep 17 00:00:00 2001 From: Megvii Engine Team Date: Fri, 16 Sep 2022 16:39:18 +0800 Subject: [PATCH] chore(mge/functional): add normalize interace to f.nn namespace GitOrigin-RevId: b42c31ca94bfde92354dff3884a9130ecb2e54a5 --- imperative/python/megengine/functional/nn.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) 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",