torch.sum()用法

torch.sum()對輸入的tensor數據的某一維度求和，一共兩種用法
１．torch.sum(input, dtype=None)
２．torch.sum(input, list: dim, bool: keepdim=False, dtype=None) → Tensor
　
input:輸入一個tensor
dim:要求和的維度，可以是一個列表
keepdim:求和之后這個dim的元素個數為１，所以要被去掉，如果要保留這個維度，則應當keepdim=True