pytorch 对一个矩阵行向量元素求和

PyTorch 中可以使用 sum 函数对一个矩阵的行向量元素进行求和，代码如下：

import torch

# 创建一个矩阵
x = torch.tensor([[1, 2, 3], [4, 5, 6]])

# 对每一行的元素求和
sums = torch.sum(x, dim=1)

print(sums)  # 输出为 tensor([ 6, 15])

希望能够帮到您！