关于tf.identity 和tf.control_dependencies

   最近在stackoverflow上看到一个问题,链接是In TensorFlow, what is tf.identity used for?。最高票的答案贴了两段代码,说明了使用tf.identity后,才会得到累加的效果,但并未解释其中原因,这篇文章做一些解释,有不对的地方还请大神指点。
首先,从tensor的定义说起,官网的解释为:A tensor is a generalization of vectors and matrices to potentially higher dimensions. tensor可以是Variable,Constant,Placeholder等等。但是,官网上还有一句话值得注意:Unlike tf.Tensor objects, a tf.Variable exists outside the context of a single session.run call.这个说明了variable的特殊性,这也就是 y = x 失效,而 y = tf.identity(x)有效的原因。同时,对于tf.control_dependencies,官网上有这么句话:control_inputs: A list of Operation or Tensor objects which must be executed or computed before running the operations defined in the context. 继续看下面的代码:

import tensorflow as tf
x = tf.Variable(0.0)
print x
x_plus_1 = tf.assign_add(x, 1)

with tf.control_dependencies([x_plus_1]):
    y = x
    print y
    #z=tf.identity(x,name='x')
init = tf.global_variables_initializer()
with tf.Session() as sess:
    sess.run(init)
    for i in range(5):
        print(sess.run(y))

运行的结果为:

<tf.Variable 'Variable:0' shape=() dtype=float32_ref>
<tf.Variable 'Variable:0' shape=() dtype=float32_ref>
2018-01-06 17:44:49.511158: I tensorflow/core/platform/cpu_feature_guard.cc:137] Your CPU supports instructions that this TensorFlow binary was not compiled to use: SSE4.2 AVX AVX2 FMA
0.0
0.0
0.0
0.0
0.0

此时y的类型为tf.Variable,不受tf.control_dependencies控制。我们把y=x换成y=x+0.0,代码变为:

import tensorflow as tf
x = tf.Variable(0.0)
print x
x_plus_1 = tf.assign_add(x, 1)

with tf.control_dependencies([x_plus_1]):
    y = x + 0.0
    print y
    #z=tf.identity(x,name='x')
init = tf.global_variables_initializer()
with tf.Session() as sess:
    sess.run(init)
    for i in range(5):
        print(sess.run(y))

再次运行代码,输出为:

<tf.Variable 'Variable:0' shape=() dtype=float32_ref>
Tensor("add:0", shape=(), dtype=float32)
2018-01-06 18:01:28.295217: I tensorflow/core/platform/cpu_feature_guard.cc:137] Your CPU supports instructions that this TensorFlow binary was not compiled to use: SSE4.2 AVX AVX2 FMA
1.0
2.0
3.0
4.0
5.0

此时,y的类型变成了tensor类型,受tf.control_dependencies的约束,生成y之前,进行了5次自加操作。最后,换成y = tf.identity(x,name='x')。代码如下:

import tensorflow as tf
x = tf.Variable(0.0)
print x
x_plus_1 = tf.assign_add(x, 1)

with tf.control_dependencies([x_plus_1]):
    y = tf.identity(x,name='x')
    print y
    #z=tf.identity(x,name='x')
init = tf.global_variables_initializer()
with tf.Session() as sess:
    sess.run(init)
    for i in range(5):
        print(sess.run(y))

运行代码,输出为:

/Users/xuyao/anaconda3/envs/python/bin/python /Users/xuyao/PycharmProjects/Test/test_tf_identity.py
<tf.Variable 'Variable:0' shape=() dtype=float32_ref>
2018-01-06 18:51:58.121838: I tensorflow/core/platform/cpu_feature_guard.cc:137] Your CPU supports instructions that this TensorFlow binary was not compiled to use: SSE4.2 AVX AVX2 FMA
Tensor("x:0", shape=(), dtype=float32)
1.0
2.0
3.0
4.0
5.0

tf.identity属于tensorflow中的一个ops,跟x = x + 0.0的性质一样,返回一个tensor,受到tf.control_dependencies的约束,所以生效。




转自:https://www.jianshu.com/p/6304b7c7896b

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值