tensorflow学习之常用函数总结:tensorflow.argmax()函数
2017-09-27 09:59
686 查看
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.csdn.net/liyaoqing/article/details/54020202
转载自:http://blog.csdn.net/zj360202/article/details/70259999
相关文章推荐
- tensorflow学习之常用函数总结:tensorflow.reduce_mean()函数
- tensorflow学习之常用函数总结:tensorflow.placeholder()函数
- tensorflow学习之常用函数总结:tensorflow.cast()函数
- tensorflow学习之常用函数总结
- TensorFlow学习笔记(十二)TensorFLow tensorBoard 总结
- tensorflow学习之常用函数总结:tensorflow官方例子中的诸如tf.reduce_mean()这类函数
- tensorflow学习之常用函数总结:tensorflow官方例子中的诸如tf.reduce_mean()这类函数
- tensorflow学习之常用函数总结:tensorflow官方例子中的诸如tf.reduce_mean()这类函数
- tensorflow学习之常用函数总结
- 【附原文:深度学习-开始Tensorflow】1.Getting Started With TensorFlow
- JS中一些常用函数学习总结
- TensorFlow学习笔记8----TensorFlow Linear Model Tutorial
- TensorFlow深度学习入门——常用函数说明
- TensorFlow和深度学习入门教程(TensorFlow and deep learning without a PhD)
- TensorFlow- tf.argmax 函数学习
- TensorFlow 学习(十五)—— tensorflow.python.platform
- Tensorflow学习笔记一:getting started with tensorflow
- 【深度学习技术】卷积神经网络常用激活函数总结
- 深度学习/神经神经网络常用激活函数总结
- 斯坦福CS20SI:基于Tensorflow的深度学习研究课程笔记,Lecture note2:TensorFlow Ops