torch.mean(tensor, dim=0)
dim 表示的是 沿着 dim 维度进行 求平均值
torch.mean()
最新推荐文章于 2024-04-10 13:28:37 发布
torch.mean(tensor, dim=0)
dim 表示的是 沿着 dim 维度进行 求平均值