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 tfVector = [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]

tf.argmax函数说明相关推荐

Tensorflow中的tf.argmax()函数
转载出处:http://www.cnblogs.com/willnote/p/6758953.html 官方API定义 tf.argmax(input, axis=None, name=None, d ...
tf.argmax函数
原文链接:https://blog.csdn.net/u013580539/article/details/79339250 关于tf.argmax,我看到网上的资料有些杂乱难以理解,所以写这篇文章. ...
tf.Graph()函数
TensorFlow是谷歌基于DistBelief进行研发的第二代人工智能学习系统,其命名来源于本身的运行原理.Tensor(张量)意味着N维数组,Flow(流)意味着基于数据流图的计算,Tensor ...
tf.argmax tf2版本
对于tf.argmax,这个函数有点奇怪,axis=0指的是计算矩阵每列的最大值索引,axis=1计算行最大值索引与numpy 相同 import tensorflow as tf import n ...
argmax函数_1.4 TensorFlow2.1常用函数
1.4 TF常用函数 tf.cast(tensor,dtype=datatype)可以进行强制类型转换. tf.reduce_min(tensor)和tf.reduce_max(tensor)将计算出 ...
tensorflow基础（2）-tf.argmax和axis
tf.argmax() 这个函数的主要功能是返回最大值所在的坐标.主要用在分类的时候,如果只是简单的输出,只是对于每一类可能性的预测的输出,但是我要要的输出必须是确定的哪一类,所以需要确定里面的最大的 ...
tensorflow学习之常用函数总结：tensorflow.argmax()函数
tensorflow.argmax()函数 tf.argmax(input, axis=None, name=None, dimension=None) 此函数是对矩阵按行或列计算最大值参数 inp ...
【Tensorflow】 tf.equal(tf.argmax(y, 1),tf.argmax(y_, 1))用法
[Tensorflow] tf.equal(tf.argmax(y, 1),tf.argmax(y_, 1))用法作用:输出正确的预测结果利用tf.argmax()按行求出真实值y_.预测值y最大 ...
tf.argmax()的详细用法
tf.argmax(data, axis=None) 用tensorflow 做 mnist分类时,用到这个接口,于是就研究了下这个接口的用法: 如果是一维数组呢? data = tf.constan ...

tf.argmax函数说明

函数说明：

参数：

用途：返回最大的那个数值所在的下标（第一个参数是矩阵，第二个参数是0或者1。0表示的是按列比较返回最大值的索引，1表示按行比较返回最大值的索引）。

代码示例：

tf.argmax函数说明相关推荐

最新文章

热门文章