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前后有两个_,

相关推荐
2zcode27 分钟前
基于Matlab融合深度学习的视频电梯乘客人数检测平台研究
深度学习·matlab·音视频
xiaoxiaoxiaolll28 分钟前
多尺度建模新范式:当神经网络融入ABAQUS,复合材料仿真迎来“物理+数据“双引擎!
人工智能·深度学习·神经网络·学习
盼小辉丶1 小时前
TensorFlow深度学习实战(29)——自监督学习(Self-Supervised Learning)
深度学习·tensorflow·自监督学习
max5006005 小时前
基于深度学习的污水新冠RNA测序数据分析系统
开发语言·人工智能·python·深度学习·神经网络
麦兜*5 小时前
Spring Boot整合PyTorch Pruning工具链,模型瘦身手术
java·pytorch·spring boot·后端·spring cloud·ai编程·剪枝
max5006006 小时前
深度学习的视觉惯性里程计(VIO)算法优化实践
人工智能·深度学习·算法
zoujiahui_20186 小时前
vscode中创建python虚拟环境的方法
ide·vscode·python
坐在地上想成仙6 小时前
计算机视觉(3)深度学习模型部署平台技术选型与全栈实践指南
人工智能·深度学习
杨荧8 小时前
基于大数据的美食视频播放数据可视化系统 Python+Django+Vue.js
大数据·前端·javascript·vue.js·spring boot·后端·python
无规则ai8 小时前
动手学深度学习(pytorch版):第一章节——引言
人工智能·pytorch·深度学习·算法·机器学习