csdn97978的博客

tf.argmax()的用法 tf.argmax()的参数形式为： tf.argmax(array / matrix，axis=0 / 1) 括号内第二个参数 axis=0 时表示按列查找 列表/矩阵 中的最大元素的索引，axis= 1时表示按行查找 列表/矩阵 中最大元素的索引 比如： import tensorflow as tf A = tf.Variable([2,6,4,8,9,1]) #array B = tf.Variable([[3,2,7], [2,9,8], [