tf.identity

  下面的程序想要执行5次循环,每次循环给x1并赋值给y,然后打印出来:

import tensorflow as tf

x = tf.Variable(0.0)
x_plus_1 = tf.assign_add(x, 1)  # 返回一个op,表示给变量x加1的操作

with tf.control_dependencies([x_plus_1]):
    y = x

init = tf.global_variables_initializer()

with tf.Session() as session:
    init.run()

    for i in range(5):
        print(y.eval())

打印出的结果都是0.0,也就是说没有达到预期的效果。这是因为y只是复制了x变量内容,并未和tensorflow图上的节点相联系,不能执行节点上的操作。进行如下修改就能实现需要的功能:

import tensorflow as tf

x = tf.Variable(0.0)
x_plus_1 = tf.assign_add(x, 1)

with tf.control_dependencies([x_plus_1]):
    y = tf.identity(x)  # 修改部分

init = tf.global_variables_initializer()

with tf.Session() as session:
    init.run()

    for i in range(5):
        print(y.eval())

也就是说,tf.identity返回了一个和x相同的的新tensor

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值