tf.argmax(vector,axis)函數的使用

1.返回值

vector爲向量，返回行或列的最大值的索引號；

vector爲矩陣，返回值是向量，返回每行或每列的最大值的索引號。

2.參數

vector爲向量或者矩陣

axis = 0 或1

0：返回vector中每列的最大值的索引號

1：返回vector中每行的最大索引號

3.例子

import numpy as np
import tensorflow as tf
a=np.array([[1,2,3]])
with tf.Session() as sess:
    print(sess.run(tf.argmax(a,1)))
>>>[2]
#表示返回行向量[1,2,3]的行最大值3的索引號是2

import numpy as np
import tensorflow as tf
a=np.array([[1,2,3]])
with tf.Session() as sess:
    print(sess.run(tf.argmax(a,0)))
>>>[0 0 0]
#表示返回行向量[1,2,3]的每列最大值，由於每列只有一個數，因此返回下標爲0 0 0 

import numpy as np
import tensorflow as tf
a=np.array([[1,2,3],[4,5,2],[-1,2,3]])
with tf.Session() as sess:
    print(sess.run(tf.argmax(a,0)))
>>>[1 1 0]
#表示返回矩陣[1,2,3;4,5,2;-1,2,3]中每列的最大值的索引號1 1 0

import numpy as np
import tensorflow as tf
a=np.array([[1],[2],[5]])
with tf.Session() as sess:
    print(sess.run(tf.argmax(a,0)))
>>>[2]
#表示返回列向量的列最大值的索引號

import numpy as np
import tensorflow as tf
a=np.array([[1],[2],[5]])
with tf.Session() as sess:
    print(sess.run(tf.argmax(a,1)))
>>>[0,0,0]
#表示返回列向量的行最大值的索引號，由於每行只有一個數，因此返回0 0 0