torch.sum()、np.sum()和sum()简要介绍

定义

• torch.sum(input1,dim,keepdim)
• np.sum(input2,axis,keepdims)
• sum(input3，b)

用法

对矩阵数据求和

• input1:输入矩阵，tensor类型
• dim:要求和的维度,默认求所有元素之和
• keepdim:求和之后，被求和的维度元素个数为1，所以要被去掉，若要保留，应keepdim=true
• input2:输入矩阵，ndarray类型
• axis:要求和的维度，同dim
• keepdims:同keepdim
• input3:输入，为列表，元组或集合(注意同矩阵的区别）
• b:在求和的基础上加上b

示例

import torch
import numpy as np

a = torch.tensor([[1,2,3],[4,5,6]])
b = np.array([[3,4,5],[6,7,8]])
c = [3,4,5,6,7,8]

print(torch.sum(a))
print(torch.sum(a,dim=0))
print(torch.sum(a,dim=1))
print(torch.sum(a,dim=1,keepdim=True))

print(np.sum(b))
print(np.sum(b,axis=0))
print(np.sum(b,axis=1))
print(np.sum(b,axis=1,keepdims=True))

print(sum(c))
print(sum(c,1))
print(sum(c,2))


>>>tensor(21)
   tensor([5, 7, 9])
   tensor([ 6, 15])
   tensor([[ 6],
           [15]])

   33
   [ 9 11 13]
   [12 21]
   [[12]
    [21]]

   33
   34
   35
​