tf.argmax 是一个非常有用的函数,它能给出某个tensor对象在某一维上的其数据最大值所在的索引值
tf.argmax(input,0) 返回input数组中按列,每列最大值所在的索引。(0-n)
tf.argmax(input,1) 返回input数组中按行,每行最大值所在的索引。(0-n)
tf.argmax 是一个非常有用的函数,它能给出某个tensor对象在某一维上的其数据最大值所在的索引值
tf.argmax(input,0) 返回input数组中按列,每列最大值所在的索引。(0-n)
tf.argmax(input,1) 返回input数组中按行,每行最大值所在的索引。(0-n)