看例子：

定義一個5×4矩陣

?我們對A矩陣所有元素求和，觀察求和的矩陣形狀：

?這里A矩陣是二維矩陣，在計算機里面也就是有兩個軸，軸0表示行，軸1表示列，求和函數A.sum()表示將A矩陣在這兩個軸方向上同時求和，所以求和后A.sum()是一個標量。

A.sum()默認是將兩個軸方向同時求和，也可以指定一個軸方向求和，下面我們指定按軸0方向求和，也就是按行求和，并觀察求和后矩陣形狀：

?可以看出求和后二維矩陣變成一維的了，按行求和，行這個維度就消失了，所以可以認為，求和就是在降維；另外提醒一下，計算機沒有行和列的概念，所以不管是按行還是按列求和降維，最終都是變成一維的。

假如我們想求和但是不想降維行不行呢，是可行的，我們用keepdim=True來實現，如下所示

可以看見，現在還是一個二維的矩陣，并沒有降維。

?

總結

以上是生活随笔為你收集整理的Pytorch中矩阵用sum()函数求和降维是怎么回事的全部內容，希望文章能夠幫你解決所遇到的問題。

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