AttributeError: module ‘torch.nn‘ has no attribute ‘module‘

bash 复制代码
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
bash 复制代码
lm=LinearModel(5)

如上,调用时报错:
init () takes 1 positional argument but 2 were given

纠错发现是少打了下划线

init前后有两个_,

相关推荐
OLOLOadsd1231 分钟前
基于NAS-FCOS的拥挤路段车辆检测系统:R50-Caffe-FPN-NASHead-GN-Head模型训练与优化_1
人工智能·深度学习
菩提树下的凡夫8 分钟前
Python 环境管理工具
开发语言·python
索荣荣24 分钟前
JavaToken实战指南:从原理到应用
开发语言·python
Albert Edison26 分钟前
【Python】函数
java·linux·python·pip
2401_8365631827 分钟前
用Python读取和处理NASA公开API数据
jvm·数据库·python
AAD555888991 小时前
基于Faster RCNN的暴力行为检测模型优化与实现_1
python
难得的我们1 小时前
超越Python:下一步该学什么编程语言?
jvm·数据库·python
工程师老罗1 小时前
Turtle库的用法
python
Network_Engineer1 小时前
从零手写RNN&BiRNN:从原理到双向实现
人工智能·rnn·深度学习·神经网络
机器学习之心1 小时前
Bayes-TCN+SHAP分析贝叶斯优化深度学习多变量分类预测可解释性分析!Matlab完整代码
深度学习·matlab·分类·贝叶斯优化深度学习