TensorFlow 基于布尔掩码选择条目（来自两个张量之一）

Question

我有三个张量，a, b, 和mask，它们的形状都一样。我想生成一个新的张量c，这样每个条目c都取自aiff 的相应条目mask为 True；否则，它取自 的相应条目b。

例子：

a = [0, 1, 2]
b = [10, 20, 30]
mask = [True, False, True]
c = [0, 20, 2]

我怎样才能做到这一点？

Answer 1

为什么不使用 tf.select(condition, t, e, name=None)

对于您的示例：

c = tf.select(mask, a, b)

有关更多详细信息tf.select，请访问Tensorflow 控制流文档