本文共 605 字,大约阅读时间需要 2 分钟。
#初始化cup的分配import osos.environ['TF_CPP_MIN_LOG_LEVEL'] = '2'"""Please note, this code is only for python 3+. If you are using python 2+, please modify the code accordingly."""import tensorflow as tfstate = tf.Variable(0, name='counter')#print(state.name)one = tf.constant(1)new_value = tf.add(state, one) #将两个变量相加update = tf.compat.v1.assign(state, new_value) #用后面的new_value更新前面的变量stateinit = tf.compat.v1.global_variables_initializer() # must have if define variablewith tf.compat.v1.Session() as sess: sess.run(init) for _ in range(3): sess.run(update) print(sess.run(state))
转载地址:http://xfkrn.baihongyu.com/