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

>>阅读原文<<