【Tensorflow】tf.argmax函数

最新推荐文章于 2024-12-13 16:49:56 发布

转载最新推荐文章于 2024-12-13 16:49:56 发布 · 96 阅读

0 ·

CC 4.0 BY-SA版权

原文链接：http://www.cnblogs.com/xqnq2007/p/7490088.html

文章标签：

#人工智能

本文详细介绍了TensorFlow中tf.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]

转载于:https://www.cnblogs.com/xqnq2007/p/7490088.html

您可能感兴趣的与本文相关的镜像