pytorch函數之nn.Linear

class torch.nn.Linear（in_features，out_features，bias = True ）[來源]

對傳入數據應用線性變換：y = A x+ b

 

參數：

in_features - 每個輸入樣本的大小

out_features - 每個輸出樣本的大小

bias - 如果設置為False，則圖層不會學習附加偏差。默認值：True

 

 

代碼：

m = nn.Linear(20, 30)

input = autograd.Variable(torch.randn(128, 20))

output = m(input)

print(output.size())

輸出：

torch.Size([128, 30])

分析：

 

output.size()=矩陣size(128,20)*矩陣size（20,30）=(128,30)