tensorflow教程之tf.argmax()用法解析

tf.argmax(a) 当只有一个参数的时候，求的是这个数组里面最大数的下标或索引

import tensorflow as tf
a=tf.argmax([1,5,90,2,3,4,5])
with tf.Session() as sess:
    print(sess.run(a))

最大的值是90，对应的索引是2，记住索引是从0开始的

 

---

如果是一个二维的矩阵呢

import tensorflow as tf
b=[[3,8,9],[2,15,0]]
a=tf.argmax(b,0)
with tf.Session() as sess:
    print(sess.run(a))

0代表y方向，1代表x方向

3与2当然是3大，3的索引为0   竖着看，

8与15 15大，索引1

9与0 9大 索引0

结果是[0,1,0] 

如果是1呢

import tensorflow as tf
b=[[3,8,9],[2,15,0]]
a=tf.argmax(b,1)
with tf.Session() as sess:
    print(sess.run(a))

 [3,8,9]最大的是9，索引为2

[2,15,0]最大的是15，索引为1