TensorFlow学习(2) 变量

1.变量(Variable)

在TensorFlow中,变量(Variable)是特殊的张量(Tensor),它的值可以是一个任何类型和形状的张量。
与其他张量不同,变量存在于单个 session.run 调用的上下文之外,也就是说,变量存储的是持久张量,当训练模型时,用变量来存储和更新参数。除此之外,在调用op之前,所有变量都应被显式地初始化过。

2.创建及使用变量

import tensorflow as tf

x = tf.Variable([1, 2])
a = tf.constant([3, 3])

sub = tf.subtract(x, a)  # 增加一个减法op
add = tf.add(x, sub)  # 增加一个加法op

# 注意变量再使用之前要再sess中做初始化,但是下边这种初始化方法不会指定变量的初始化顺序
init = tf.global_variables_initializer()#全局变量初始化
with tf.Session() as sess:
    sess.run(init)
    print(sess.run(sub))
    print(sess.run(add))


# 创建一个名字为‘counter’的变量 初始化0
state = tf.Variable(0, name='counter')
new_value = tf.add(state, 1)  # 创建一个op,作用是使state加1
update = tf.assign(state, new_value)  # 赋值op,不能直接用等号赋值,作用是 state = new_value,借助tf.assign()函数
init = tf.global_variables_initializer()#全局变量初始化

with tf.Session() as sess:
    sess.run(init)
    print(sess.run(state))
    for _ in range(5):#循环5次
        sess.run(update)
        print(sess.run(state))

注意:

  • 所有变量在session使用前必须初始化,可以使用全局变量初始化语句
    在session创建后,也需运行init,进行初始化操作。(详见上述代码)

init = tf.global_variables_initializer()
sess.run(init)

  • 使某个变量等于另一个变量,不能直接使用等号,而是需要使用tf.assign()函数,使其相等。

update = tf.assign(state, new_value) # 作用是 state = new_value

3.结果展示

在这里插入图片描述

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值