tf.argmax函数说明

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

来源：CSDN

作者：like_study_cat

链接：https://blog.csdn.net/like_study_cat/article/details/103973607