tf.random.set_seed
是 TensorFlow 的一个函数,用于设置随机数生成器的种子。
函数的签名如下:
tf.random.set_seed(seed)
参数说明:
seed
:一个整数,表示随机数生成器的种子。
这个函数的作用是设置 TensorFlow 中的随机数生成器的种子,以确保每次生成的随机数序列是确定的。这对于调试和测试非常有用,因为我们可以使用固定的随机种子来重现实验结果。
请注意,tf.random.set_seed
函数只会影响 TensorFlow 中的随机数生成器,不会影响 Python 自身的随机数生成器或其他库的随机数生成器。如果你需要设置 Python 自身的随机数种子,请使用 numpy.random.seed
或 random.seed
函数。
下面是一个使用示例:
import tensorflow as tf
# 设置随机种子
tf.random.set_seed(123)
# 生成随机张量
x = tf.random.uniform([2, 2])
print(x)
# 再次生成随机张量
y = tf.random.uniform([2, 2])
print(y)
在这个示例中,我们首先使用 tf.random.set_seed
函数设置了随机种子为123。然后,我们使用 tf.random.uniform
函数生成了两个形状为 [2, 2]
的随机张量 x
和 y
。由于我们设置了固定的随机种子,每次运行代码时生成的随机数序列都是相同的,因此 x
和 y
的值是确定的。