import torch
import torch.nn as nn
class LinearModel(nn.Module):
def _init_(self,ndim):
super(LinearModel,self)._init_()
self.ndim=ndim
self.weight=nn.Parameter(torch.randn(ndim,1))#定义权重
self.bias=nn.Parameter(torch.randn(1)) #定义偏置
def forward(self,x):
# y = wx +b
return x.mm(self.weight)+self.bias
lm=LinearModel(5)
如上，调用时报错：
init() takes 1 positional argument but 2 were given
纠错发现是少打了下划线
init前后有两个_,文章来源地址https://www.toymoban.com/news/detail-563612.html
文章来源：https://www.toymoban.com/news/detail-563612.html
到了这里，关于AttributeError: module ‘torch.nn‘ has no attribute ‘module‘的文章就介绍完了。如果您还想了解更多内容，请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章，希望大家以后多多支持TOY模板网！
