tf.argmax(vector, 1)

原创于 2019-01-05 12:20:04 发布 · 190 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#argmax #tensorflow

TensorFlow 专栏收录该内容

8 篇文章

订阅专栏

本文详细解析了 TensorFlow 中的 tf.argmax 函数，该函数用于找出输入 tensor 中最大值的索引。通过具体示例，展示了当输入为向量或矩阵时，函数如何返回相应的最大值索引。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

tf.argmax(vector, 1)：返回的是vector中的最大值的索引号，如果vector是一个向量，那就返回一个值，如果是一个矩阵，那就返回一个向量，这个向量的每一个维度都是相对应矩阵行的最大值元素的索引号。

import tensorflow as tf
A = tf.constant([[1,2,3,4,5]])
B = tf.constant([[1,2,3],[6,5,4]])
with tf.Session() as sess:
    print(sess.run(tf.argmax(A,1)))
    print(sess.run(tf.argmax(B,1)))

输出
[4]
[2 0]