python编程中,各种随机种子seed设置总结

2022-12-21 15:55:44 浏览数 (1)

首先导入库:

代码语言:javascript复制
# 导入模块
import random
import numpy as np
import tensorflow as tf
import torch
import time

下面先展示python内置random函数numpy中的random函数tensorflowpytorch中常见的seed使用方式(注:pytorch仅以CPU为例):

代码语言:javascript复制
seed = 1

random.seed(seed)
np.random.seed(seed)
tf.random.set_seed(seed)
torch.manual_seed(seed)

list = [1,2,3,4,5,6,7,8,9]

a = random.sample(list,5)
b = np.random.randn(5)
c = tf.random.normal([5])
d = torch.randn(5)

print(time.strftime("%Y-%m-%d %H:%M:%S", time.localtime()))
print('python内置输出:',a)
print('*' * 60)
print('numpy输出:',b)
print('*' * 60)
print('tensorflow输出:',c)
print('*' * 60)
print('pytorch输出',d)


# 第一次运行输出:
2021-01-17 17:51:36
python内置输出: [3, 2, 9, 1, 4]
************************************************************
numpy输出: [ 1.62434536 -0.61175641 -0.52817175 -1.07296862  0.86540763]
************************************************************
tensorflow输出: tf.Tensor([-1.1012203   1.5457517   0.383644   -0.87965786 -1.2246722 ], shape=(5,), dtype=float32)
************************************************************
pytorch输出 tensor([ 0.6614,  0.2669,  0.0617,  0.6213, -0.4519])

# 第二次运行输出:
2021-01-17 17:52:10
python内置输出: [3, 2, 9, 1, 4]
************************************************************
numpy输出: [ 1.62434536 -0.61175641 -0.52817175 -1.07296862  0.86540763]
************************************************************
tensorflow输出: tf.Tensor([-1.1012203   1.5457517   0.383644   -0.87965786 -1.2246722 ], shape=(5,), dtype=float32)
************************************************************
pytorch输出 tensor([ 0.6614,  0.2669,  0.0617,  0.6213, -0.4519])

0 人点赞