tf.argmax(input, axis=None, name=None, dimension=None)
此函數是對矩陣按行或列計算最大值

參數
input：輸入Tensor
axis：0表示按列，1表示按行
name：名稱
dimension：和axis功能一樣，默認axis取值優先。新加的字段
返回：Tensor 一般是行或列的最大值下標

import tensorflow as tf a=tf.Variable(tf.random_uniform([3,4],minval=-1,maxval=1)) b=tf.argmax(input=a,axis=0) c=tf.argmax(input=a,dimension=1) #此處用dimesion或用axis是一樣的 sess = tf.Session() sess.run(tf.initialize_all_variables()) print(sess.run(a)) #[[ 0.04261756 -0.34297419 -0.87816691 -0.15430689] # [ 0.18663144 0.86972666 -0.06103253 0.38307118] # [ 0.84588599 -0.45432305 -0.39736366 0.38526249]] print(sess.run(b)) #[2 1 1 2] print(sess.run(c)) #[0 1 0]

總結

以上是生活随笔為你收集整理的tf.argmax的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。