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)