仅选择两个输入流中的一个

Question

我有一个图表，应该在每个会话调用中使用两个输入流之一进行选择。如果设置了标志，则应使用stream_1，如果未设置，则应使用stream_2。 tf.cond可以完成此任务，但正如文档所述，https://www.tensorflow.org/api_docs/python/tf/cond，无论是否实际使用它们，都会计算输入。由于stream_1计算量很大，我宁愿不使用这样的解决方案。还有另一种方法可以完成同样的事情，当没有设置标志时，根本不计算stream_1吗？