增加as

Author: pengcheng888

python/infinicore/nn/__init__.py

@@ -1,5 +1,5 @@
 from infinicore.nn import functional
 from infinicore.nn.modules import *  # noqa: F403
-from infinicore.nn.parameter import Parameter
+from infinicore.nn.parameter import InfiniCoreParameter as Parameter
 
 __all__ = ["functional", "Parameter"]

python/infinicore/nn/modules/__init__.py

@@ -1,4 +1,4 @@
-from .container import ModuleList
-from .module import Module
+from .container import InfiniCoreModuleList as ModuleList
+from .module import InfiniCoreModule as Module
 
 __all__ = ["ModuleList", "Module"]

python/infinicore/nn/modules/container.py

@@ -9,10 +9,7 @@
 from itertools import chain
 from typing import Iterator, List, Optional, Sequence, TypeVar, Union
 
-from .module import Module
-
-__all__ = ["ModuleList"]
-
+from .module import InfiniCoreModule as Module
 
 # Define type variable for module compatibility (supports InfiniCoreModule)
 ModuleType = TypeVar("ModuleType", bound=Union["Module"])
@@ -189,6 +186,3 @@ def __dir__(self) -> List[str]:
         # Filter out numeric keys to avoid cluttering dir() output
         keys = [key for key in keys if not key.isdigit()]
         return keys
-
-
-ModuleList = InfiniCoreModuleList

python/infinicore/nn/modules/module.py

@@ -33,9 +33,7 @@
 import infinicore
 
 from ...tensor import Tensor
-from ..parameter import Parameter
-
-__all__ = ["Module"]
+from ..parameter import InfiniCoreParameter as Parameter
 
 _EXTRA_STATE_KEY_SUFFIX = "_extra_state"
 T = TypeVar("T", bound="InfiniCoreModule")
@@ -857,6 +855,3 @@ def _apply(self, fn, recurse=True):
 
     def to(self, *args, **kwargs):
         raise KeyError("not support")
-
-
-Module = InfiniCoreModule

python/infinicore/nn/parameter.py

@@ -32,6 +32,3 @@ def __deepcopy__(self, memo):
 
     def __reduce_ex__(self, proto):
         raise ValueError("not supported!")
-
-
-Parameter = InfiniCoreParameter