tensorflow: tf.assign探究

来源:互联网 发布:arrayslice php 编辑:程序博客网 时间:2024/06/07 16:42

tf.assign(ref, value, validate_shape=None, use_locking=None, name=None)

  Defined in tensorflow/python/ops/state_ops.py.

  将 value 赋值给 ref,并输出 ref 

  这使得需要使用复位值的连续操作变简单

  Defined in tensorflow/python/framework/tensor_shape.py.

Args Annotations 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 :

  Same as “ref”. Returned as a convenience for operations that want to use the new value after the variable has been reset.



原创粉丝点击