tensorflow如何在每个节点增加阈值判断？

就是当节点计算完激活函数之后，要增加一个阈值判断，超过了，会向下个节点输出1，没超过就输出0这样，不知道怎么弄，跪求大佬们教教我

one = tf.ones_like(label)
zero = tf.zeros_like(label)
label = tf.where(label <0.5, x=zero, y=one)
这个好像可以，参考：https://blog.csdn.net/xxy0118/article/details/85999155