TensorFlow插曲--tf.argmax函数-白红宇

TensorFlow插曲--tf.argmax函数

阅读量：4147 次

发布时间：2019-05-25

本文共 722 字，大约阅读时间需要 2 分钟。

转自：

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

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

参数

input：输入Tensor

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

name：名称

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

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

例：

[java]
 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]  