Python tf.keras.utils.set_random_seed用法及代码示例

设置程序的所有随机种子(Python、NumPy 和 TensorFlow)。

用法

tf.keras.utils.set_random_seed(
    seed
)

您可以使用此实用程序使几乎所有 Keras 程序都具有完全确定性。一些限制适用于涉及网络通信(例如参数服务器分布)的情况，这会产生额外的随机源，或者涉及某些非确定性 cuDNN 操作。

调用此实用程序等效于以下内容：

import random
import numpy as np
import tensorflow as tf
random.seed(seed)
np.random.seed(seed)
tf.random.set_seed(seed)