功能：查找最大值的索引
input：数据；
axis：定义按照某一条轴进行查找，如：
axis=0 按x轴查找最大值的索引，axis=1 按y轴查找最大值的索引，以此类推2,3,4,...即为多维张量时
name：定义操作名称
注意：tf.argmax()与numpy.argmax()作用相同。
argmin()正好与argmax作用相反，它用于查找最小值索引
temsorflow使用笔记（自用）
原文：https://www.cnblogs.com/freeyouth/p/11600179.html
