提交 fd802e97 编写于 作者: M Megvii Engine Team

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

GitOrigin-RevId: eb90506535e19f09a30bcf6ed9ba5a56653281a6
上级 a640eeff
......@@ -35,7 +35,6 @@ __all__ = [
"conv_transpose2d",
"dot",
"dropout",
"embedding",
"indexing_one_hot",
"interpolate",
"leaky_relu",
......
......@@ -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
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册