在 PyTorch 中，可以使用 torchsummary 库来实现对 PyTorch 模型的结构及参数统计的输出，其可以方便我们查看每层输入、输出的维度以及参数数量等信息。
安装 torchsummary 库：
pip install torchsummary
使用方法如下：
import torch
from torchsummary import summary
# 定义一个模型
model = torch.nn.Sequential(
)
# 打印模型所有层的参数统计
summary(model, (3, 32, 32))
其中，model 是需要查看的模型，(3, 32, 32) 表示模型的输入维度，即 C = 3，H = 32，W = 32。运行后可以看到所有层输入、输出的维度以及参数数量等信息。
有个大坑
AttributeError: ‘xxx’ object has no attribute ‘size’
AttributeError: ‘int’ object has no attribute ‘size’
类似错误文章来源：https://www.toymoban.com/news/detail-563799.html
pip uninstall torchsummary
pip install torch-summary
解决问题文章来源地址https://www.toymoban.com/news/detail-563799.html
到了这里，关于Pytorch模型如何查看每层输入维度输出维度的文章就介绍完了。如果您还想了解更多内容，请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章，希望大家以后多多支持TOY模板网！
