一般定义一个linear层的时候,写法为nn.linear(in_features,out_features)
具体形式为:
y=wx+b
weight=Parameter(torch.Tensor(out_features,in_features))
bias=Parameter(torch.Tensor(out_features))
--------------------- 
作者:小小码农JACK 
来源:CSDN 
原文:https://blog.csdn.net/haoshan4783/article/details/88803737 
版权声明:本文为博主原创文章,转载请附上博文链接!