Machine learning 在tensorflow中，如何输入>；0.5变成张量中的1？_Machine Learning_Tensorflow_Neural Network_Deep Learning

Machine learning 在tensorflow中，如何输入>；0.5变成张量中的1？

machine-learning tensorflow neural-network deep-learning

Machine learning 在tensorflow中，如何输入>；0.5变成张量中的1？,machine-learning,tensorflow,neural-network,deep-learning,Machine Learning,Tensorflow,Neural Network,Deep Learning,例如，a=[0,0.2,0.8] 我想让它成为a=[0,0.2,1] 我搜索了很长时间，但我只找到了tf.bool_mask，它只能得到布尔张量，我不知道下一步该怎么办我知道，至少在numpy中，这很容易，但如何在tensorflow中实现这一点？您可以使用和ops： 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]

例如，

a=[0,0.2,0.8]

我想让它成为

a=[0,0.2,1]

我搜索了很长时间，但我只找到了

tf.bool_mask

，它只能得到布尔张量，我不知道下一步该怎么办

我知道，至少在numpy中，这很容易，但如何在tensorflow中实现这一点？

您可以使用和ops：

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]