在Pytorch中计算一个特定维度的4D张量的平均值

import torch

# 创建一个4D张量
tensor = torch.randn(2, 3, 4, 5)

# 指定要计算平均值的维度
dim = 2

# 计算指定维度的平均值
mean_tensor = torch.mean(tensor, dim=dim)

print(mean_tensor)

import torch

# 创建一个4D张量
tensor = torch.randn(2, 3, 4, 5)

# 指定要计算平均值的维度
dim = 2

# 计算指定维度的平均值
mean_tensor = torch.mean(tensor, dim=dim)

# 打印平均值
print(mean_tensor)