pytorch矩阵求和

可以使用PyTorch中的sum函数来对矩阵进行求和。具体实现如下：

import torch

# 生成一个3x3的随机矩阵
mat = torch.rand(3, 3)

# 对矩阵的所有元素进行求和
result = torch.sum(mat)

print(result)

如果需要对矩阵的某个维度进行求和，可以使用dim参数指定需要求和的维度。例如，对于一个3x3的随机矩阵，如果想对每一列进行求和，可以这样实现：

import torch

# 生成一个3x3的随机矩阵
mat = torch.rand(3, 3)

# 对矩阵的每一列进行求和
result = torch.sum(mat, dim=0)

print(result)

这将输出一个长度为3的一维张量，表示矩阵每一列的求和结果。