tf.identity()是否会创建完整的深层副本?

时间:2017-01-30 07:03:45

标签: tensorflow

我经常使用以下模式来管理控制流程:

with tf.get_default_graph().control_dependencies([c_op]):
     h_state = tf.identity(h_state)

但是,我担心tf.identity()可能会复制传递给它的数据,这不是我想要的。有人可以确认它是否创建副本?

1 个答案:

答案 0 :(得分:2)

tf.identity()操作的implementation会将其输入转发到其输出而不进行深层复制。但是,如果将tf.identity()操作固定到与产生其输入的操作不同的设备上,则会发生深度复制