diff --git a/mindspore/common/seed.py b/mindspore/common/seed.py index bd797d69e5..bfed1d11ae 100644 --- a/mindspore/common/seed.py +++ b/mindspore/common/seed.py @@ -86,53 +86,53 @@ def set_seed(seed): >>> >>> # 3. If neither global seed nor op seed is set, mindspore.ops.composite.random_ops and >>> # mindspore.nn.probability.distribution will choose a random seed: - >>> c1 = C.uniform((1, 4)) # C1 - >>> c2 = C.uniform((1, 4)) # C2 + >>> c1 = C.uniform((1, 4), 1.0, 2.0) # C1 + >>> c2 = C.uniform((1, 4), 1.0, 2.0) # C2 >>> Rerun the program will get different results: - >>> c1 = C.uniform((1, 4)) # C3 - >>> c2 = C.uniform((1, 4)) # C4 + >>> c1 = C.uniform((1, 4), 1.0, 2.0) # C3 + >>> c2 = C.uniform((1, 4), 1.0, 2.0) # C4 >>> >>> # 4. If global seed is set, but op seed is not set, mindspore.ops.composite.random_ops and >>> # mindspore.nn.probability.distribution will caculate a seed according to global seed and >>> # default op seed. Each call will change the default op seed, thus each call get different >>> # results. >>> set_seed(1234) - >>> c1 = C.uniform((1, 4)) # C1 - >>> c2 = C.uniform((1, 4)) # C2 + >>> c1 = C.uniform((1, 4), 1.0, 2.0) # C1 + >>> c2 = C.uniform((1, 4), 1.0, 2.0) # C2 >>> # Rerun the program will get the same results: >>> set_seed(1234) - >>> c1 = C.uniform((1, 4)) # C1 - >>> c2 = C.uniform((1, 4)) # C2 + >>> c1 = C.uniform((1, 4), 1.0, 2.0) # C1 + >>> c2 = C.uniform((1, 4), 1.0, 2.0) # C2 >>> >>> # 5. If both global seed and op seed are set, mindspore.ops.composite.random_ops and >>> # mindspore.nn.probability.distribution will caculate a seed according to global seed and >>> # op seed counter. Each call will change the op seed counter, thus each call get different >>> # results. >>> set_seed(1234) - >>> c1 = C.uniform((1, 4), seed=2) # C1 - >>> c2 = C.uniform((1, 4), seed=2) # C2 + >>> c1 = C.uniform((1, 4), 1.0, 2.0, seed=2) # C1 + >>> c2 = C.uniform((1, 4), 1.0, 2.0, seed=2) # C2 >>> Rerun the program will get the same results: >>> set_seed(1234) - >>> c1 = C.uniform((1, 4), seed=2) # C1 - >>> c2 = C.uniform((1, 4), seed=2) # C2 + >>> c1 = C.uniform((1, 4), 1.0, 2.0, seed=2) # C1 + >>> c2 = C.uniform((1, 4), 1.0, 2.0, seed=2) # C2 >>> >>> # 6. If op seed is set but global seed is not set, 0 will be used as global seed. Then >>> # mindspore.ops.composite.random_ops and mindspore.nn.probability.distribution act as in >>> # condition 5. - >>> c1 = C.uniform((1, 4), seed=2) # C1 - >>> c2 = C.uniform((1, 4), seed=2) # C2 - >>> #Rerun the program will get the same results: - >>> c1 = C.uniform((1, 4), seed=2) # C1 - >>> c2 = C.uniform((1, 4), seed=2) # C2 + >>> c1 = C.uniform((1, 4), 1.0, 2.0, seed=2) # C1 + >>> c2 = C.uniform((1, 4), 1.0, 2.0, seed=2) # C2 + >>> # Rerun the program will get the same results: + >>> c1 = C.uniform((1, 4), 1.0, 2.0, seed=2) # C1 + >>> c2 = C.uniform((1, 4), 1.0, 2.0, seed=2) # C2 >>> >>> # 7. Recall set_seed() in the program will reset numpy seed and op seed counter of >>> # mindspore.ops.composite.random_ops and mindspore.nn.probability.distribution. >>> set_seed(1234) >>> np_1 = np.random.normal(0, 1, [1]).astype(np.float32) # A1 - >>> c1 = C.uniform((1, 4), seed=2) # C1 + >>> c1 = C.uniform((1, 4), 1.0, 2.0, seed=2) # C1 >>> set_seed(1234) >>> np_2 = np.random.normal(0, 1, [1]).astype(np.float32) # still get A1 - >>> c2 = C.uniform((1, 4), seed=2) # still get C1 + >>> c2 = C.uniform((1, 4), 1.0, 2.0, seed=2) # still get C1 """ if not isinstance(seed, int): raise TypeError("The seed must be type of int.")