一般定义一个linear层的时候,写法为nn.linear(in_features,out_features)
具体形式为:
y=wx+b
weight=Parameter(torch.Tensor(out_features,in_features))
bias=Parameter(torch.Tensor(out_features))
pytorch中nn.linear
最新推荐文章于 2024-06-26 20:09:38 发布
一般定义一个linear层的时候,写法为nn.linear(in_features,out_features)
具体形式为:
y=wx+b
weight=Parameter(torch.Tensor(out_features,in_features))
bias=Parameter(torch.Tensor(out_features))