torch.nn.Linear()函数理解

函数：class torch.nn.Linear（in_features，out_features，bias = True） 源码： 从init函数中可以看出Linear中包含四个属性： 1）in_features: 上层神经元个数【每个输入样本的大小】 2）out_features： 本层神经元个数【每个输出样本的大小】 3）weight：权重，形状[out_features , in_feat

>>阅读原文<<