Tensorflow 多个输入到图形

Question

这可能是微不足道的。我有一个模型需要一些出列张量X;

X = tf.Tensor(...)
yPred = model(X,...)

现在我有时想喂另一个张量Z；

Z = tf.placeholder(...)
yPredZ = model(Z,...)

如何在不重新定义子图的情况下做到这一点？

Answer 1

TensorFlow 中的馈送机制允许您为 任何 张量（不仅仅是 tf.placeholder() 张量）馈送值，只要它们的形状和元素类型匹配即可。

因此，如果x和z的形状相同，你应该可以写成：

x = ...  # Some dequeued `tf.Tensor`. 
yPred = model(x, ...)

# ...

sess.run(yPred, feed_dict={x: ...})

在某些情况下，您可能希望将具有不同形状的张量提供给 x（通常是不太具体的形状，例如具有不同的批量大小维度）。在这些情况下，您可以使用tf.placeholder_with_default() 创建一个占位符，当您不馈送它时，其值默认为x：

x = ...  # Some dequeued `tf.Tensor`.

# For example, a shape of `None` means that the shape is completely unconstrained.
# In practice, you will probably want to constrain at least the rank of the
# placeholder to match the rank of `x`.
x_placeholder = tf.placeholder_with_default(x, shape=None)

yPred = model(x, ...)

# ...

sess.run(yPred, feed_dict={x_placeholder: ...})