tf.argmax() 函数

函数说明：

tf.argmax( input, axis=None, name=None, dimension=None, output_type=tf.int64)

参数：

input:输入矩阵
axis:默认为None
name:默认为None
dimension:默认为None
output_type:默认类型为int64

用途：返回最大的那个数值所在的下标（第一个参数是矩阵，第二个参数是0或者1。0表示的是按列比较返回最大值的索引，1表示按行比较返回最大值的索引）。

代码示例：

import tensorflow as tf
Vector = [1,1,2,5,3] #定义一个向量
X = [[1,3,2],[2,5,8],[7,5,9]] #定义一个矩阵
with tf.Session() as sess:
a = tf.argmax(Vector, 0)
b = tf.argmax(X, 0)
c = tf.argmax(X, 1)
print(sess.run(a))
print(sess.run(b))
print(sess.run(c))

运行结果：

3
[2 1 2]
[1 2 2]
————————————————

原文链接：https://blog.csdn.net/kdongyi/article/details/82390394