tf.where(a,b,c)函数:
当a为true时,tf.where(a,b,c)函数会选择b值输出
当a为false时,tf.where(a,b,c)函数会选择c值输出
div = tf.div(sub, labels)
ones = tf.ones_like(div)
zeros = tf.zeros_like(div)
result = tf.where(div >= 0.05, ones, zeros)
equal = tf.cast(result, tf.bool) #将返回结果转化为布尔型
代码解析:
功能:判断div有多少个大于0.05的数,
结果:如果大于0.05,就返回1,即ones。
小于0.05就返回0, 即zeros 。
result 就是一个0和1的tensor 。