对于 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)
Thanks.
当然,你可以使用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])>
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)