tf.argmax()

时间:2022-06-19
本文章向大家介绍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