tensorflow中argmax、equal、cast、reduce_sum、reduce_mean等函数的用法
tf.argmax、tf.equal、tf.cast、tf.reduce_sum、tf.reduce_mean用法a=tf.argmax([[0,1,0,0],[1,0,0,0],[0,0,0,1]],axis=1)#tf.argmax(a,axis=1)a是一个列向量,其中的元素是行向量。最后得到的是最大值的下标,按行b=tf.equal([0,2,3],[0,1,3])#tf.equal(a,