Browse Source

!6163 modify __all__ notes

Merge pull request !6163 from baiyangfan/add_notes
tags/v1.0.0
mindspore-ci-bot Gitee 5 years ago
parent
commit
b9c996484e
2 changed files with 6 additions and 4 deletions
  1. +4
    -4
      mindspore/common/__init__.py
  2. +2
    -0
      mindspore/nn/__init__.py

+ 4
- 4
mindspore/common/__init__.py View File

@@ -20,12 +20,12 @@ from .parameter import Parameter, ParameterTuple
from .tensor import MetaTensor, Tensor, RowTensor, SparseTensor
from .seed import set_seed, get_seed

__all__ = [

__all__ = dtype.__all__
__all__.extend([
"MetaTensor", "Tensor", "RowTensor", "SparseTensor", # tensor
'ms_function', # api
'Parameter', 'ParameterTuple', # parameter
"dtype",
"set_seed", "get_seed" # random seed
]

__all__.extend(dtype.__all__)
])

+ 2
- 0
mindspore/nn/__init__.py View File

@@ -18,6 +18,7 @@ Neural Networks Cells.
Pre-defined building blocks or computing units to construct Neural Networks.
"""
from . import layer, loss, optim, metrics, wrap, probability, sparse
from .learning_rate_schedule import *
from .cell import Cell, GraphKernel
from .layer import *
from .loss import *
@@ -34,5 +35,6 @@ __all__.extend(optim.__all__)
__all__.extend(metrics.__all__)
__all__.extend(wrap.__all__)
__all__.extend(sparse.__all__)
__all__.extend(learning_rate_schedule.__all__)

__all__.sort()

Loading…
Cancel
Save