zl程序教程

您现在的位置是:首页 >  其它

当前栏目

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

输入输出 形状 NN linear weight 问题 神经元
2023-09-11 14:22:51 时间

 weight形状为[out_features, in_features]

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

 

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