tf.assign函数的用法

tf.assign(
   
ref,
    value
,
    validate_shape
=None,
    use_locking
=None,
    name
=None
)
功能:   这个操作输出一个张量,它在赋值之后保持“REF”的新值。这使得更容易重置值时去使用链式操作。 例子:    import tensorflow as tf     import numpy as np     A = tf.Variable(tf.ones([2,2]), dtype=tf.float32)       with tf.Session() as sess:           sess.run(tf.global_variables_initializer())           print("原始的A ")         print(sess.run(A))           sess.run(tf.assign(A,np.zeros([2,2])))           print("现在的A ")         print(sess.run(A))   输出:    原始的A      [[1. 1.]      [1. 1.]]     现在的A      [[0. 0.]      [0. 0.]] 参数解释:
  • ref: A mutable Tensor. Should be from a Variable node. May be uninitialized.
  • value: A Tensor. Must have the same type as ref. The value to be assigned to the variable.
  • validate_shape: An optional bool. Defaults to True. If true, the operation will validate that the shape of 'value' matches the shape of the Tensor being assigned to. If false, 'ref' will take on the shape of 'value'.
  • use_locking: An optional bool. Defaults to True. If True, the assignment will be protected by a lock; otherwise the behavior is undefined, but may exhibit less contention.
  • name: A name for the operation (optional).

猜你喜欢

转载自blog.csdn.net/qq_29023939/article/details/80388936