Browse Source

feat(mge): do not export F.nn.embedding

GitOrigin-RevId: eb90506535
release-1.1
Megvii Engine Team 4 years ago
parent
commit
fd802e978f
2 changed files with 1 additions and 2 deletions
  1. +0
    -1
      imperative/python/megengine/functional/nn.py
  2. +1
    -1
      imperative/python/megengine/module/embedding.py

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

@@ -35,7 +35,6 @@ __all__ = [
"conv_transpose2d", "conv_transpose2d",
"dot", "dot",
"dropout", "dropout",
"embedding",
"indexing_one_hot", "indexing_one_hot",
"interpolate", "interpolate",
"leaky_relu", "leaky_relu",


+ 1
- 1
imperative/python/megengine/module/embedding.py View File

@@ -10,7 +10,7 @@ from typing import Optional


import numpy as np import numpy as np


from ..functional import embedding as embedding_func
from ..functional.nn import embedding as embedding_func
from ..tensor import Parameter from ..tensor import Parameter
from . import init from . import init
from .module import Module from .module import Module


Loading…
Cancel
Save