torch.nn.Linear解释


 

 

torch.nn.Linear的作用是对输入向量进行矩阵的乘积和加法。y=x(A)转置+b。这点类似于全连接神经网络的的隐藏层。in_feature代表输入神经元的个数。out_feature代表输出神经元的个数。bias为False不参与训练。如果为True则参与训练。

x = torch.randn(20)  # 输入的维度是(20)
m = torch.nn.Linear(20, 1)  # 20,1是指输入维度、输出维度 神经网络又20个输入神经元,1个输出神经元。
output = m(x)
print('m.weight.shape:\n ', m.weight.shape)
print('m.bias.shape:\n', m.bias.shape)
print('output.shape:\n', output.shape)
输出结果:

m.weight.shape:
torch.Size([1, 20])//运算时,权值需要转置。
m.bias.shape:
torch.Size([1])//只有一个神经元故bias只有一个。
output.shape:
torch.Size([1])//一个神经元只有一个输出值。


免责声明!

本站转载的文章为个人学习借鉴使用,本站对版权不负任何法律责任。如果侵犯了您的隐私权益,请联系本站邮箱yoyou2525@163.com删除。



 
粤ICP备18138465号  © 2018-2025 CODEPRJ.COM