Tensorflow 2中tf.variable的条件赋值

Question

对于 numpy，我们有

threshold = 3
a = np.array([1,2,3,4,5,6])
a[a>=3] = 199

# a is [1, 2, 199, 199, 199, 199]

如何在tensorflow 2写类似的代码

b = tf.Variable(a)

谢谢。

Answer 1

当然，您可以使用tf.where 有条件地设置值：

b = tf.Variable(a)
tf.where(b >= 3, 199, b)
# <tf.Tensor: shape=(6,), dtype=int64, numpy=array([  1,   2, 199, 199, 199, 199])>