如何在张量流中设置随机种子

问题描述

有人可以告诉我如何在张量流中设置随机种子吗? 就像在numpy中一样:

np.random.seed(seed = some_number)

我尝试过:

with tf.Session() as sess:
tf.set_random_seed(seed = 42) 

,但似乎不起作用。 谢谢!

解决方法

要设置全局随机种子:

tf.random.set_seed(seed)

参数seed是您的some_number

来自https://www.tensorflow.org/api_docs/python/tf/random/set_seed