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})

猜你喜欢

转载自blog.csdn.net/weixin_34613450/article/details/80725099
今日推荐