Browse Source

chore(mge/functional): add normalize interace to f.nn namespace

GitOrigin-RevId: b42c31ca94
release-1.11.1
Megvii Engine Team 2 years ago
parent
commit
213f40437c
1 changed files with 2 additions and 1 deletions
  1. +2
    -1
      imperative/python/megengine/functional/nn.py

+ 2
- 1
imperative/python/megengine/functional/nn.py View File

@@ -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",


Loading…
Cancel
Save