torch.mean(tensor, dim=0)
dim 表示的是 沿着 dim 维度进行 求平均值
torch.mean()
最新推荐文章于 2024-07-17 15:11:52 发布
torch.mean(tensor, dim=0)
dim 表示的是 沿着 dim 维度进行 求平均值