相当于Tensorflow中的theano.tensor.switch

Question

Tensorflow中是否有与theano.tensor.switch(cond, ift, iff)完全相同的功能？

Answer 1

在TensorFlow中，您可以使用tf.cond（https://www.tensorflow.org/api_docs/python/tf/cond）。文档中有一些例子。

编辑：正如你所提到的，这个操作不是元素明智的，那么它等价于tf.where（https://www.tensorflow.org/api_docs/python/tf/where）