大家在tensorflow碰到一些,比如reduce_sum, contact 函數的時候,對其中axis 參數位 是0,1,2,-1等數字可能比較模糊。
有些說法是參數0表示跨行操作,也就是縱向操作,
1是跨列操作,也就是橫向操作
但這些概念只能用在二維列表中,三維和更高維度無法理解。
所以,我對這個參數的理解是,axis這個數字代表的就是操作的維度,0表示第一維,1是第二維,-1是最後一維。
舉個例子:
在這個二維tensor中,第一維是行,所以參數爲0的時候,是表示在行上進行操作,也就是跨行操作。
第二維是列,準確的說應該是每行裏面的各自的元素,所以參數爲1時,是對每行裏各自的計算,得出的是【3,3】
因爲是二維的,所以當參數爲-1時,效果和參數爲1 相同。即最後一維。
本人學識有限,如有錯誤望指正。