python - 函数 control_dependencies 有什么作用？

Question

我想要一个例子来说明 tf.control_dependencies 函数的使用。 .例如，我想创建两个张量 X 和 Y 并且如果它们相等，则执行或打印某些内容。

import tensorflow as tf

session = tf.Session()

X = tf.constant(5)
Y = tf.constant(50)

with tf.control_dependencies([tf.assert_equal(X, Y)]):
    print('X and Y are equal!')

在上面的代码中，X 显然不等于Y。在这种情况下，tf.control_dependencies 在做什么？

Answer 1

control_dependencies 不是条件。这是一种将依赖项添加到您在 with block 中创建的任何操作的机制。更具体地说，您在 control_dependencies 的参数中指定的内容确保在您在 with block 中定义的任何内容之前进行评估。

在您的示例中，您没有在 with block 中添加任何 (TensorFlow) 操作，因此该 block 什么也不做。

This answer有一个如何使用 control_dependencies 的示例，它用于确保在评估 batchnorm 操作之前发生分配。