tf.equal()和tf.cast()用法

在看代码的时候看到人家这么写，然后自己写个demo做浅显的理解
tf.equal(a,b)
:对于单个变量，如果a==b,则返回true,否则返回false;
对于数组，会迭代的比较a[i]和b[i]，对应相等的则返回true,否则返回false

tf.cast(a, dtype=tf.float32)
:将a的类型转换成tf.float32

import tensorflow as tf
a = [1, 2, 3]
b = [1, 5, 3]
sess = tf.Session()
with sess.as_default():
    acc = sess.run(tf.equal(a, b))
    print(acc)　　
    print(sess.run(tf.reduce_mean(tf.cast(acc, dtype=tf.float32))))

运行结果如下图所示：

代码解释：
假设a是输入的y_hat, b是预测的y
tf.equal(a, b)会比较每一个y和y_hat 是否相等
tf.cast(acc, dtype=tf.float32)将比较的结果转换成tf.float32类型
tf.reduce_mean()求所有tf.float32类型结果的均值