Nigel
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

modify seed API

This commit is contained in:
lilei 2021-06-24 10:46:11 +08:00
parent 8388ced9ee
commit b31e763d7e
1 changed files with 8 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
mindspore/common/seed.py
View File

@ -157,7 +157,10 @@ def set_seed(seed):
def get_seed(): def get_seed():
""" """
Get global random seed. Get global seed.
Returns:
Integer. The global seed.
""" """
return _GLOBAL_SEED return _GLOBAL_SEED
@ -168,8 +171,11 @@ def _truncate_seed(seed):
Args: Args:
seed (int): The seed to be truncated. seed (int): The seed to be truncated.
Returns:
Integer. The seed with MAXINT32.
""" """
return seed % _MAXINT32 # Truncate to fit into 32-bit integer return seed % _MAXINT32
def _update_seeds(op_seed, kernel_name): def _update_seeds(op_seed, kernel_name):