pytorch求数据平均值保持维度

要求求平均值并保持维度，可以使用`torch.mean`函数，并且指定`dim参数为要保持的维度。下面是一个示例代码：

import torch

# 创建一个3维张量
x = torch.tensor([[1, 2, 3], [4, 5, 6], [7, 8, 9]])

# 求平均值并保持维度
mean = torch.mean(x, dim=0)

print(mean)

输出结果为：

tensor([4., 5., 6.])

在上面的示例中，我们创建了一个3维张量`x`，然后使用`torch.mean`函数求取了每一列的平均值，并且通过指定`dim=0`参数来保持维度。最后将结果打印出来。