首页
学习
活动
专区
工具
TVP
发布
精选内容/技术社群/优惠产品,尽在小程序
立即前往

tensorflow: tf.assign探究

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.



下一篇
举报
领券