import tensorflow as tf
input = [1, 3, 5, 7]
output = ([[1, 2],[3, 5]])
a = tf.argmax(input, 0) # 0表是按行
b = tf.argmax(output, 1) # 1 表示按列
with tf.Session() as sess:
print("输出input最大值的索引下标:", sess.run(a))
print("输出output最大值的索引下标,这是一个二维数组:",sess.run(b))
#rerurn:
3
[1 1]
【tensorflow】tf-argmax()
最新推荐文章于 2022-02-12 12:38:06 发布