1. 程式人生 > >tensorflow對tf.argmax()的理解

tensorflow對tf.argmax()的理解

在tensorflow中,axis=1對列進行操作,axis=0對行進行操作

  • 例如,argmax中axis=1就是對(每行中)不同列元素進行比較,得到最大值。
a = [[0, 0, 0, 1, 0, 0],
 [0, 0, 1, 0, 0, 0],
 [0, 1, 0, 0, 0, 0],
 [0, 0, 0, 0, 0, 1]]
 
 # axis=1, 每行的不同列進行比較,取對應最大值的索引
tf.argmax(a, 1) => [3 2 1 5]

# axis=0, 每列的不同行進行比較,去對應最大值的索引
tf.argmax(a, 0) => [0 2 1 0 0 3]