参考地址:https://www.jianshu.com/p/cde4a33fa129

torch初学
按照图中的代码敲一遍(注意view层的使用方法,以及最后计算参数数量的方式。)
torch初学

参数数量的计算结果:
torch初学
参数结果中的【6】、【16】、【120】、【84】、【10】全是指Bias所带来的参数量。

相关文章: