TensorFlow中assign函数说明

2018-09-01 18:10 更新

tf.assign


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

定义在:tensorflow/python/ops/state_ops.py

参见指南:变量>变量帮助函数

通过将 "value" 赋给 "ref" 来更新 "ref".

此操作输出在赋值后保留新值 "ref" 的张量.这使得更易于链接需要使用重置值的操作.

ARGS:

  • ref:一个可变的张量.应该来自变量节点.节点可能未初始化.
  • value:张量.必须具有与 ref 相同的类型.是要分配给变量的值.
  • validate_shape:一个可选的 bool.默认为 True.如果为 true, 则操作将验证 "value" 的形状是否与分配给的张量的形状相匹配;如果为 false, "ref" 将对 "值" 的形状进行引用.
  • use_locking:一个可选的 bool.默认为 True.如果为 True, 则分配将受锁保护;否则, 该行为是未定义的, 但可能会显示较少的争用.
  • name:操作的名称(可选).

返回:

一个在赋值完成后将保留 "ref" 的新值的张量.

tf.assign_add


assign_add (
ref ,
value ,
use_locking = None ,
name = None
)

定义在:tensorflow/python/ops/state_ops.py

参见指南:变量>变量帮助函数

通过向 "ref" 添加 "value" 来更新 "ref".

此操作在更新完成后输出 "ref".这使得更易于链接需要使用重置值的操作.

ARGS:

  • ref:一个可变的张量.必须是下列类型之一:float32,float64,int64,int32,uint8,uint16,int16,int8,complex64,complex128,qint8,quint8,qint32,half.应该来自变量节点.
  • value:张量.必须有相同的类型 ref,要添加到变量的值.
  • use_locking:默认为 False;如果为真, 加法将被锁保护;否则,该行为是未定义的,但可能会显示较少的争用.
  • name:操作的名称(可选).

返回:

与“REF”相同.返回是为了希望在更新变量后使用新值的操作的便利.

tf.assign_sub


assign_sub (
ref ,
value ,
use_locking = None ,
name = None
)

定义在:tensorflow/python/ops/state_ops.py

参见指南:变量>变量帮助函数

通过从 "ref" 中减去 "value" 来更新 "ref".

此操作在更新完成后输出 "ref",这使得更易于链接需要使用重置值的操作.

ARGS:

  • ref:一个可变的张量.必须是下列类型之一:float32,float64,int64,int32,uint8,uint16,int16,int8,complex64,complex128,qint8,quint8,qint32,half.应该来自变量节点.
  • value:张量.必须有相同的类型 ref.要减去变量的值.
  • use_locking:一个可选的 bool,默认为 False.如果为 True,减法将被锁定保护,否则行为是未定义的,但可能表现出较少的争用. 
  • name:操作的名称(可选).

返回:

与“REF”相同.返回是为了使得在更新变量后要使用新值的操作的便利.

以上内容是否对您有帮助:
在线笔记
App下载
App下载

扫描二维码

下载编程狮App

公众号
微信公众号

编程狮公众号

pFad - Phonifier reborn

Pfad - The Proxy pFad of © 2024 Garber Painting. All rights reserved.

Note: This service is not intended for secure transactions such as banking, social media, email, or purchasing. Use at your own risk. We assume no liability whatsoever for broken pages.


Alternative Proxies:

Alternative Proxy

pFad Proxy

pFad v3 Proxy

pFad v4 Proxy