From fd802e978f01e2668887fc2a01745dbd20d8e2dc Mon Sep 17 00:00:00 2001
From: Megvii Engine Team <megengine@megvii.com>
Date: Fri, 9 Oct 2020 22:24:43 +0800
Subject: [PATCH] feat(mge): do not export F.nn.embedding

GitOrigin-RevId: eb90506535e19f09a30bcf6ed9ba5a56653281a6
---
 imperative/python/megengine/functional/nn.py    | 1 -
 imperative/python/megengine/module/embedding.py | 2 +-
 2 files changed, 1 insertion(+), 2 deletions(-)

diff --git a/imperative/python/megengine/functional/nn.py b/imperative/python/megengine/functional/nn.py
index 4a00e836..48056bcb 100644
--- a/imperative/python/megengine/functional/nn.py
+++ b/imperative/python/megengine/functional/nn.py
@@ -35,7 +35,6 @@ __all__ = [
     "conv_transpose2d",
     "dot",
     "dropout",
-    "embedding",
     "indexing_one_hot",
     "interpolate",
     "leaky_relu",
diff --git a/imperative/python/megengine/module/embedding.py b/imperative/python/megengine/module/embedding.py
index 7b7db1f3..c9d8da9c 100644
--- a/imperative/python/megengine/module/embedding.py
+++ b/imperative/python/megengine/module/embedding.py
@@ -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