tensorflow学习之常用函数总结：tensorflow.argmax()函数

最新推荐文章于 2025-10-13 13:46:31 发布

转载最新推荐文章于 2025-10-13 13:46:31 发布 · 1.5k 阅读

文章标签：

#tensorflow #函数

Tensorflow学习笔记专栏收录该内容

5 篇文章

订阅专栏

本文详细介绍了 TensorFlow 中的 argmax 函数及其使用方法。argmax 函数用于找到输入张量中最大值的索引位置，可通过设置 axis 参数来指定操作的方向。文章通过示例代码展示了如何使用该函数。

tensorflow.argmax()函数

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

此函数是对矩阵按行或列计算最大值

参数
input：输入Tensor

axis：0表示按列，1表示按行

name：名称

dimension：和axis功能一样，默认axis取值优先。新加的字段

返回：Tensor 一般是行或列的最大值下标向量

例：
    import tensorflow as tf  
      
    a=tf.get_variable(name='a',  
                      shape=[3,4],  
                      dtype=tf.float32,  
                      initializer=tf.random_uniform_initializer(minval=-1,maxval=1))  
    b=tf.argmax(input=a,axis=0)  
    c=tf.argmax(input=a,dimension=1)   #此处用dimesion或用axis是一样的  
    sess = tf.InteractiveSession()  
    sess.run(tf.initialize_all_variables())  
    print(sess.run(a))  
    #[[ 0.04261756 -0.34297419 -0.87816691 -0.15430689]  
    # [ 0.18663144  0.86972666 -0.06103253  0.38307118]  
    # [ 0.84588599 -0.45432305 -0.39736366  0.38526249]]  
    print(sess.run(b))  
    #[2 1 1 2]  
    print(sess.run(c))  
    #[0 1 0]  
上述实例引用： http://blog.youkuaiyun.com/liyaoqing/article/details/54020202

转载自：http://blog.youkuaiyun.com/zj360202/article/details/70259999