tf.argmax()
tf.argmax() 返回矩阵横列或者纵列的最大值的坐标，取决于第二个参数
参数：
tf.argmax(
input,
axis=None,//0是纵列 1是横列
name=None,
dimension=None,
output_type=tf.int64
)使用案例：
import tensorflow as tf
x = tf.constant([[0, 2, 1], [2, 1, 3]])
sess = tf.Session()
print(sess.run(tf.argmax(x, 0)))//[1 0 1]
print(sess.run(tf.argmax(x, 1)))//[1 2]tf.argmax(vector, 1)：返回的是vector中的最大值的索引号，如果vector是一个向量，那就返回一个值，如果是一个矩阵，那就返回一个向量，这个向量的每一个维度都是相对应矩阵行的最大值元素的索引号。
import tensorflow as tf
import numpy as np
A = [[1,3,4,5,6]]
B = [[1,3,4], [2,4,1]]
with tf.Session() as sess:
print(sess.run(tf.argmax(A, 1)))
print(sess.run(tf.argmax(B, 1)))输出：
[4]
[2 1]参考：https://blog.csdn.net/uestc_c2_403/article/details/72232807
本文参与 腾讯云自媒体分享计划，分享自作者个人站点/博客。
原始发表：2018年12月30日，如有侵权请联系 cloudcommunity@tencent.com 删除
评论
登录后参与评论
推荐阅读