一般定义一个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
版权声明:本文为博主原创文章,转载请附上博文链接!