方法

直接将参数传入即可。如下代码，有三个以上的输入也是以此类推

summary(model,first_input,second_input,device='cpu')

还有问题可以去这里查看torch-summary

相关推荐
高子歌
 【pytorch】评估计算 PyTorch 模型的 FLOPs 和参数量
高子歌 34 0 0
未定义变量
 pytorch之网络参数统计 torchstat & torchsummary
未定义变量 71 0 0
elvinyang
 神经网络的计算量和参数量估计
elvinyang 86 0 0
绣文字
 pytorch统计模型的参数量
绣文字 14 0 0
伢赞
 多输入多输出 | MATLAB实现CNN-GRU卷积门控循环单元多输入多输出
伢赞 9 0 0
嚯霍嚯
 Excel快速输入时间
嚯霍嚯 204 0 0
waaagh
 多输入多输出 | MATLAB实现LSSVM最小二乘支持向量机多输入多输出
waaagh 40 0 0
ITWYY
 多输入多输出 | MATLAB实现CNN-BiGRU卷积双向门控循环单元多输入多输出
ITWYY 28 0 0
朱小落
 iOS UITextView输入时禁止换行
朱小落 160 0 0
大柚子top
 各个语言模型大小，参数量等统计
大柚子top 81 0 0

精彩评论（0）