nn.linear(weight、bias和输入输出神经元的形状问题)


 weight形状为[out_features, in_features]

简单的说就是,在定义时使用的是[out_features, in_features],而在单层线性神经网络计算时使用的是weight的转置矩阵。

 

https://blog.csdn.net/dss_dssssd/article/details/83537765


免责声明!

本站转载的文章为个人学习借鉴使用,本站对版权不负任何法律责任。如果侵犯了您的隐私权益,请联系本站邮箱yoyou2525@163.com删除。



 
粤ICP备18138465号  © 2018-2025 CODEPRJ.COM