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