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

 weight形状为[out_features, in_features]

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

 

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

posted @ 2022-01-25 17:56  Tomorrow1126  阅读(762)  评论(0编辑  收藏  举报