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


 weight形状为[out_features, in_features]

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

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

相关