import tensorflow as tf
#0表示的是按列比较返回最大值的索引,1表示按行比较返回最大值的索引
Vector = [1,1,2,5,3] #定义一个向量
X = [[1,3,2],[2,5,8],[7,5,9]] #定义一个矩阵
with tf.Session() as sess:
a = tf.argmax(Vector, 0)
b = tf.argmax(X, 0)
c = tf.argmax(X, 1)
print(sess.run(a))
print(sess.run(b))
print(sess.run(c))
'''
3
[2 1 2]
[1 2 2]
'''
#tf.argmax(vector, 1):返回的是vector中的最大值的索引号
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]
'''