對(duì)于tf.argmax,這個(gè)函數(shù)有點(diǎn)奇怪，axis=0指的是計(jì)算矩陣每列的最大值索引，axis=1計(jì)算行最大值索引
與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] ------------------------------分割線------------------------------

總結(jié)

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

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。