動手學習深度學習 | python函數學習

(1):維度dim，保持原有維度keepdim 下面通過圖像的形式直觀的展示了這兩個參數的作用。 X = torch.tensor([[1, 2, 3], [4, 5, 6]]) print(X.sum(dim=0, keepdim=True)) # dim爲0，按照相同的列求和，並在結果中保留列特徵 print(X.sum(dim=1, keepdim=True)) # dim爲1，按照相同的行

>>阅读原文<<