一、tf.argmax(input,axis,name=None) : 返回input中最大值的索引index。
其中axis=0时表示比较每一列的元素,返回最大元素所在行号(即第一维度);
axis=1表示比较每一行的元素, 返回最大元素所在列号(即第二维度);
二、tf.argmin(input,axis,name=None) : 返回input中最小值的索引index。
其中axis=0时表示比较每一列的元素,返回最小元素所在行号(即第一维度);
其中axis=1时表示比较每一行的元素,返回最小元素所在列号(即第二维度);