tensorflow argmax

最新推荐文章于 2021-08-13 15:12:58 发布

转载最新推荐文章于 2021-08-13 15:12:58 发布 · 214 阅读

人工智能专栏收录该内容

13 篇文章

订阅专栏

本文详细介绍了 TensorFlow 中的 tf.argmax 函数用法。该函数用于找出输入 tensor 中最大值的位置索引，支持向量和矩阵等多种输入形式。通过具体实例展示了如何在不同维度上应用此函数。

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

[python]view plain copy
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)))  