在 tensorflow 中一些函數的 axis 的理解

大家在tensorflow碰到一些,比如reduce_sum, contact 函數的時候,對其中axis 參數位 是0,1,2,-1等數字可能比較模糊。

有些說法是參數0表示跨行操作,也就是縱向操作,

1是跨列操作,也就是橫向操作

但這些概念只能用在二維列表中,三維和更高維度無法理解。

所以,我對這個參數的理解是,axis這個數字代表的就是操作的維度,0表示第一維,1是第二維,-1是最後一維。

舉個例子:

在這個二維tensor中,第一維是行,所以參數爲0的時候,是表示在行上進行操作,也就是跨行操作。

第二維是列,準確的說應該是每行裏面的各自的元素,所以參數爲1時,是對每行裏各自的計算,得出的是【3,3】

因爲是二維的,所以當參數爲-1時,效果和參數爲1 相同。即最後一維。

本人學識有限,如有錯誤望指正。

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章