Tensorflow：seed用法详解

Posted 2022-08-13 -柚子皮-

 

##### file1 #####
import tensorflow as tf

# tf.set_random_seed(1234)
# a = tf.Variable(tf.random_uniform([2, 2], 0, 1), name="aaa")
a = tf.Variable(tf.random_uniform([2, 2], 0, 1, seed=1234, name="aaa"))
# a = tf.Variable(tf.random_uniform([2, 2], 0, 1, name="aaa"))

with tf.Session() as sess1:
    sess1.run(tf.initialize_all_variables())
    sess1.run(a)
    print(sess1.run(a))  # generates 'A1'
    print(sess1.run(a))  # generates 'A2'
    saver = tf.train.Saver()
    path = saver.save(sess=sess1, save_path="./model.ckpt")

##### file2 #####
import tensorflow as tf

# tf.set_random_seed(1234)
saver = tf.train.import_meta_graph(meta_graph_or_file="./model.ckpt.meta")

with tf.Session() as sess:
    saver.restore(sess=sess, save_path="./model.ckpt")
    # tensor_a = tf.get_default_graph().get_tensor_by_name(name="aaa:0")
    tensor_a = sess.graph.get_tensor_by_name(name="aaa:0")
    print(sess.run(tensor_a))

[tf.random.set_random_seed]

from: -柚子皮-

ref: