Tensorflow–tf.assign()详解

最新推荐文章于 2021-04-09 16:13:10 发布

诚朴求食

最新推荐文章于 2021-04-09 16:13:10 发布

阅读量2k

点赞数 2

CC 4.0 BY-SA版权

分类专栏： python 机器学习深度学习文章标签： TensorFlow 机器学习深度学习人工智能 Python

本文链接：https://blog.csdn.net/fengchi863/article/details/102560329

tf中的tf.assign()函数非常容易被人误解，如果不是非常透彻理解tensorflow graph和op的概念的话，一不小心就会计算错误。

下面先来看一下源代码中的注释：

def assign(ref, value, validate_shape=None, use_locking=None, name=None):
  """Update 'ref' by assigning 'value' to it.
 
  This operation outputs a Tensor that holds the new value of 'ref' after
    the value has been assigned. This makes it easier to chain operations
    that need to use the reset value.
 
  Args:
    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).
 
  Returns:
    A `Tensor` that will hold the new value of 'ref' after
      the assignment has completed.
  """
  if ref.dtype._is_ref_dtype:
    return gen_state_ops