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",
"dot",
"dropout",
"embedding",
"indexing_one_hot",
"interpolate",
"leaky_relu",


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

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

import numpy as np

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


Loading…
Cancel
Save