Tensorflow中 tf.argmax()的用法

本文介绍了TensorFlow库中的tf.argmax()函数,该函数用于查找列表或矩阵中的最大值索引。通过设置参数axis,可以指定按列(axis=0)或按行(axis=1)查找最大值的索引位置。
摘要由CSDN通过智能技术生成

tf.argmax()的用法

tf.argmax()的参数形式为:

tf.argmax(array / matrix,axis=0 / 1)

括号内第二个参数 axis=0 时表示按列查找 列表/矩阵 中的最大元素的索引,axis= 1时表示按行查找 列表/矩阵 中最大元素的索引

比如:

import tensorflow as tf

A = tf.Variable([2,6,4,8,9,1])  #array
B = tf.Variable([[3,2,7],
				[2,9,8],
				[6,4,1]])       #matrix

A_1 = tf.cast(tf.argmax(A,0),tf.int64)   #类型转化
B_1 = tf.cast(tf.argmax(B,0),tf.int64)
B_2 = tf.cast(tf.argmax(B,1),tf.int64)
				
init = tf.global_variables_initializer()

with tf.Session() as sess:
	sess.run(init)
	print(A_1.eval())
	print(B_1.eval())
	print(B_2.eval())

结果:

4       #列表中最大数的索引       
[2 1 1]   #矩阵中每一列最大数的索引
[2 1 0]   #矩阵中每一行最大数的索引
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值