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.argmax()函数相关推荐

Tensorflow学习笔记6：解决tensorflow训练过程中GPU未调用问题
Tensorflow学习笔记6:解决tensorflow训练过程中GPU未调用问题参考文章: (1)Tensorflow学习笔记6:解决tensorflow训练过程中GPU未调用问题 (2)http ...
tensorflow学习之常用函数总结：tensorflow.placeholder()函数
tensorflow.placeholder()函数 tensorflow.placeholder(dtype, shape=None, name=None) 此函数可以理解为形参,用于定义过程,在执 ...
TensorFlow学习笔记（1）--TensorFlow简介，常用基本操作
要将深度学习更快且更便捷地应用于新的问题中,选择一款深度学习工具是必不可少的步骤. TensorFlow是谷歌于2015年11月9日正式开源的计算框架.TensorFlow计算框架可以很好地支持深度学 ...
TensorFlow 学习（十五）—— tensorflow.python.platform
tensorflow.python.platform 下的常用工具类和工具函数:tensorflow/tensorflow/python/platform at master · tensorflow ...
tensorflow学习笔记九：将 TensorFlow 移植到 Android手机，实现物体识别、行人检测和图像风格迁移详细教程
2017/02/23 更新贴一个TensorFlow 2017开发者大会的Mobile专题演讲移动和嵌入式TensorFlow 这里面有重点讲到本文介绍的三个例子,以及其他的移动和嵌入式方面的TF ...
tensorflow学习之常用函数总结：tensorflow官方例子中的诸如tf.reduce_mean()这类函数
前言 tensorflow官网给的例子用到了很多函数,然后并没有具体说明,还要自己去翻文档,有些函数是很常用的,下面来一一总结. 正文一,tensorflow中有一类在tensor的某一维度上求值的 ...
tensorflow学习之常用函数总结：tensorflow.reduce_mean()函数
前言 tensorflow官网给的例子用到了很多函数,然后并没有具体说明,还要自己去翻文档,有些函数是很常用的,下面来一一总结. 正文 Tensorflow中有一类在tensor的某一维度上求值的函数 ...
tensorflow学习之常用函数总结：tensorflow.cast()函数
tensorflow.cast()类型转换函数 tf.cast(x, dtype, name=None) 此函数是类型转换函数参数 x:输入 dtype:转换目标类型 nam ...
TensorFlow学习笔记（十三）TensorFLow 常用Optimizer 总结
这里主要是各种优化器,以及使用.因为大多数机器学习任务就是最小化损失,在损失定义的情况下,后面的工作就交给优化器啦. 因为深度学习常见的是对于梯度的优化,也就是说,优化器最后其实就是各种对于梯度下降算 ...

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

tensorflow.argmax()函数

tensorflow学习之常用函数总结：tensorflow.argmax()函数相关推荐

最新文章

热门文章