pytorch 求和函数

以下是PyTorch中的求和函数的示例：

1. 使用`torch.sum()`函数对张量进行逐元素求和：

import torch

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

result = torch.sum([a, b, c])
print(result)  # 输出：tensor(45)

2. 使用`torch.sum()`函数对指定轴方向进行求和：

import torch

matrix = torch.tensor([[1, 2, 3],
                       [4, 5, 6],
                       [7, 8, 9]])

# 按轴0方向求和，即按行求和
result = torch.sum(matrix, dim=0)
print(result)  # 输出：tensor([12, 15, 18])

# 按轴1方向求和，即按列求和
result = torch.sum(matrix, dim=1)
print(result)  # 输出：tensor([ 6, 15, 24])