Tensorflow之基础函数解析1

1.tf.argmax

tf.argmax就是返回最大的那个数值所在的下标,axis=0:按列比较；axis=1：按行比较

 

2. tf.equal 

tf.equal(A, B)是对比这两个矩阵或者向量的相等的元素，如果是相等的那就返回True，反之返回False，返回的值的矩阵维度和A是一样的；

3.tf.cast

tf.cast(x, dtype, name=None) 
        将x的数据格式转化成dtype.例如，原来x的数据格式是bool， 
        那么将其转化成float以后，就能够将其转化成0和1的序列;

 

4.tf.reduce_mean       

tf.reduce_mean：求均值

5.accuracy.eval

accuracy = tf.reduce_mean(tf.cast(correct_prediction, 'float'))

accuracy.eval:在一个Seesion里面“评估”tensor的值（其实就是计算）

accuracy.eval(feed_dict={X:X_train, Y:Y_train})