對于tf.argmax,這個函數有點奇怪，axis=0指的是計算矩陣每列的最大值索引，axis=1計算行最大值索引
與numpy 相同

import tensorflow as tf import numpy as npa=np.array([[2,4,5,7],[9,3,6,2]]) print('-'*30+'分割線'+'-'*30) print(a) print('-'*30+'分割線'+'-'*30)a1=tf.argmax(a,axis=0)print('tf.argmax(a,axis=0)=',a1) print('-'*30+'分割線'+'-'*30) a1=np.argmax(a,axis=0) print('np.argmax(a,axis=0)=',a1) print('-'*30+'分割線'+'-'*30)a1=tf.argmax(a,axis=1)print('tf.argmax(a,axis=1)=',a1) print('-'*30+'分割線'+'-'*30)a1=np.argmax(a,axis=1)print('np.argmax(a,axis=1)=',a1) print('-'*30+'分割線'+'-'*30) ------------------------------分割線------------------------------ [[2 4 5 7][9 3 6 2]] ------------------------------分割線------------------------------ tf.argmax(a,axis=0)= tf.Tensor([1 0 1 0], shape=(4,), dtype=int64) ------------------------------分割線------------------------------ np.argmax(a,axis=0)= [1 0 1 0] ------------------------------分割線------------------------------ tf.argmax(a,axis=1)= tf.Tensor([3 0], shape=(2,), dtype=int64) ------------------------------分割線------------------------------ np.argmax(a,axis=1)= [3 0] ------------------------------分割線------------------------------

總結

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

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