TensorFlow函数tf.argmax()是一个常用的函数，它的作用是从一个张量中找出最大值的索引，返回一个张量，其中包含索引信息。tf.argmax()函数有两个参数，第一个参数是一个张量，第二个参数是一个整数，表示按照指定的维度求最大值索引。
使用方法
tf.argmax()函数的使用方法如下：
# 定义一个张量 x = tf.constant([[1, 2, 3], [4, 5, 6]]) # 调用tf.argmax()函数，按照第一个维度求最大值索引 result = tf.argmax(x, axis=0) # 运行结果 # result ==> [1, 1, 1]
上面的代码求出的结果是[1, 1, 1]，表示在每一列中，最大值的索引都是1。
如果需要按照第二个维度求最大值索引，可以把axis参数设置为1：
# 调用tf.argmax()函数，按照第二个维度求最大值索引 result = tf.argmax(x, axis=1) # 运行结果 # result ==> [2, 2]
上面的代码求出的结果是[2, 2]，表示在每一行中，最大值的索引都是2。
除了按照指定的维度求最大值索引，tf.argmax()函数还可以求出整个张量中最大值的索引，只需要把axis参数设置为-1即可：
# 调用tf.argmax()函数，求出整个张量中最大值的索引 result = tf.argmax(x, axis=-1) # 运行结果 # result ==> 5
上面的代码求出的结果是5，表示整个张量中最大值的索引是5。
可以看出，tf.argmax()函数可以根据用户的需求，按照指定的维度求最大值索引，也可以求出整个张量中最大值的索引。