看例子：

定義一個(gè)5×4矩陣

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

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

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

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

假如我們想求和但是不想降維行不行呢，是可行的，我們用keepdim=True來(lái)實(shí)現(xiàn)，如下所示

可以看見(jiàn)，現(xiàn)在還是一個(gè)二維的矩陣，并沒(méi)有降維。

?

總結(jié)

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

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