TF计算操作

导入TensorFlow包

import tensorflow as tf
tf.argmax() 返回最大值的位置
eg: t1 = [2,1,8]
    t2 = [[3,4,8],[2,9,4]]
    tf.argmax(t1,1)  ==> 2
    tf.argmax(t2,1)  ==>[2,1]
tf.matmul()  矩阵乘法
tf.reduce_sum()    求和
tf.reduce_mean()  求平均值
tf.equal()  判断预测类别是否正确
tf.cast()  值类型转换

eg: p 类型为tf.int32
    a = tf.cast(p,tf.float32)将p转成float32类型的值
tf.run()  tf.eval() 的区别
在使用t.eval()等价于:tf.get_default_session().run(t)
tf.run(t,feed_dict{}) ==> tf.eval(feed_dict{})

你可能感兴趣的:(TF计算操作)