tf.argmax()

tf.argmax()函数返回某一行或某一列的第一个最大值的位置,
所以在卷积网络里面就可以把tf.argmax()用到最后一层，提前把每种分类和位置对应好。
tf.argmax(input, axis=None, name=None, dimension=None)这是这个函数的完整参数。
input参数可以是一个列表、array矩阵或者是一个张量
当axis=0，返回每一列最大值所在的位置
当axis=1，返回每一行最大值所在位置
当axis=0时，举例如下：

import tensorflow as tf
temp = tf.constant([[1, 2, 3], [3, 2, 1], [3, 4, 5], [5, 4, 3]])
with tf.Session() as sess:
    sess.run(tf.global_variables_initializer())
print(sess.run(tf.argmax(temp, axis=0)))

输出为

[3 2 2]

当axis=1时

import tensorflow as tf
temp = tf.constant([[1, 2, 3], [3, 2, 1], [3, 4, 5], [5, 4, 3]])
with tf.Session() as sess:
    sess.run(tf.global_variables_initializer())
print(sess.run(tf.argmax(temp, axis=0)))

输出为

[2 0 2 0]