tf.random.set_seed

100 阅读1分钟

tf.random.set_seed 是 TensorFlow 的一个函数,用于设置随机数生成器的种子。

函数的签名如下:

tf.random.set_seed(seed)

参数说明:

  • seed:一个整数,表示随机数生成器的种子。

这个函数的作用是设置 TensorFlow 中的随机数生成器的种子,以确保每次生成的随机数序列是确定的。这对于调试和测试非常有用,因为我们可以使用固定的随机种子来重现实验结果。

请注意,tf.random.set_seed 函数只会影响 TensorFlow 中的随机数生成器,不会影响 Python 自身的随机数生成器或其他库的随机数生成器。如果你需要设置 Python 自身的随机数种子,请使用 numpy.random.seedrandom.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] 的随机张量 xy。由于我们设置了固定的随机种子,每次运行代码时生成的随机数序列都是相同的,因此 xy 的值是确定的。