torch.mean(input, dim, keepdim=False, out=None) → Tensor
**其中,keepdim的参数很重要,如果使用默认的False,则默认降维,如果使用True,就不进行降维
**
torch.mean
最新推荐文章于 2025-07-14 00:16:35 发布
torch.mean(input, dim, keepdim=False, out=None) → Tensor
**其中,keepdim的参数很重要,如果使用默认的False,则默认降维,如果使用True,就不进行降维
**