如何在Tensorflow中对张量中的元素设置阈值操作？

Question

当我得到网络的输出时，它是一个大小如[batch_size，height，weight]的张量。内容是概率。我想要做的是为张量设置一个阈值，并进行二值化处理。那么我应该对张量做些什么呢？

Answer 1

您可以使用tf.clip_by_value：

import tensorflow as tf

a = tf.random_uniform(shape=(2, 3, 3), minval=-1, maxval=3)
b = tf.clip_by_value(a, 0, 1)

with tf.Session() as sess:
    A, B = sess.run([a, b])
    print A, '\n'
    print B

这里1以上的所有内容都将变为1，0以下的所有内容都将为0.其他内容将保持不变。

另请查看此answer，其执行类似操作但将所有内容转换为0或1：