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

相关推荐
Wishell201524 分钟前
日拱一卒之Python与matlab的内存读取区别
pytorch
许泽宇的技术分享29 分钟前
AgentFramework: 安全最佳实践
python·安全·flask
小鸡吃米…29 分钟前
Python - JSON
开发语言·python·json
AI即插即用31 分钟前
即插即用系列 | CVPR 2024 RMT:既要全局感受野,又要 CNN 的局部性?一种拥有显式空间先验的线性 Transformer
人工智能·深度学习·神经网络·目标检测·计算机视觉·cnn·transformer
渡我白衣33 分钟前
导论:什么是机器学习?——破除迷思,建立全景地图
人工智能·深度学习·神经网络·目标检测·microsoft·机器学习·自然语言处理
smile_Iris40 分钟前
Day 45 简单CNN
人工智能·深度学习·cnn
我的xiaodoujiao41 分钟前
使用 Python 语言 从 0 到 1 搭建完整 Web UI自动化测试学习系列 34--基础知识 9--文件上传功能
前端·python·测试工具·ui·pytest
咸鱼加辣44 分钟前
【python面试】Python 的 lambda
javascript·python·算法
感谢地心引力1 小时前
【AI】加入AI绘图的视频封面快速编辑器
人工智能·python·ai·ffmpeg·音视频·pyqt·gemini
我一定会有钱1 小时前
pytest测试框架基础
python·单元测试·自动化·pytest