例如，a = [0,0.2,0.8]

我想将其变为a = [0,0.2,1]

我搜索了很长时间，只找到了tf.bool_mask，它只能得到一个布尔张量，我不知道下一步该怎么做。

我知道至少在NumPy中，这很容易，但在TensorFlow中该怎么做呢？

回答：

你可以使用tf.where()和tf.ones_like()操作来实现这一点：

a = tf.constant([0.0, 0.2, 0.8])
result = tf.where(a > 0.5, tf.ones_like(a), a)  # result is [0.0, 0.2, 1.0]