一.函数: argmax( input, axis=None, name=None, dimension=None, output_type=tf.int64 ) 简单的说,tf.argmax就是返回最大的那个数值所在的下标. 二.用法 [code] import numpy as np test = np.array([[1, 2, 3], [2, 3, 4], [5, 4, 3], [8, 7, 2]]) np.argmax(test,0) #输出:array([3, 3, 1], dtyp…