Pytorch笔记(1-5)

张量

张量可以理解为numpy里面的矩阵,可以进行矩阵的操作

张量初始化

python 复制代码
x_data = torch.tensor([[1,2],[2,3]])

模型的建立

首先建立一个单元,在pytorch里面,一个单元就是神经网络的一个节点,前向传播。

一般以一个类定义

python 复制代码
class LearnModel(torch.nn.Module):
    def __init__(self):
        super(LinearModel,self).__init__()# 必须要有
        self.linear = torch.nn.Linear(1,1)# 接受两个参数一个是输入数据的维度,一个是输出数据的维度,这个函数就是求出参数w和b再顺便做好相乘加b
    def forward(self,x):
        y_pred = self.linear(x)# 前向传播
        return y_pred

损失函数和参数更新

python 复制代码
criterion = torch.nn.MSELoss(size_average = True)
# 损失函数,均方根损失函数,接受两个参数,上面这个为True时求均值
optimizer = torch.optim.SGD(model.parameters(), lr = 0.01)
# 这是一个优化器,就是更新参数的,第一个参数就是你要更新哪些参数,第一个函数可以选择所有的参数,lr就是learn_rate学习率

线性回归

python 复制代码
model = LinearModel()
for epoch in range(100):
    y_pred = model(x_data) # 前向传播获得y的预测值
    loss = criterion(y_pred, y_data) # 求出代价
    optimizer.zero_grad() # 清零梯度
    loss.backward() # 自动计算梯度
    optimizer.step() # 更新w和b的值
相关推荐
我的golang之路果然有问题1 分钟前
案例速成GO+Socket,个人笔记
开发语言·笔记·后端·websocket·学习·http·golang
roc-ever1 分钟前
用Python做有趣的AI项目 6:AI音乐生成器(LSTM Melody Generator)
人工智能·python·lstm
我的golang之路果然有问题3 分钟前
快速了解Go+rpc
开发语言·经验分享·笔记·rpc·golang·go
song_ly0015 分钟前
《软件测试52讲》学习笔记:如何设计一个“好的“测试用例?
笔记·学习·测试用例
xiaohanbao098 分钟前
day10 python机器学习全流程实践
人工智能·python·学习·机器学习·信息可视化·pandas
layneyao9 分钟前
神经辐射场(NeRF)技术解析:3D重建与虚拟世界的未来
人工智能·3d
kyle~20 分钟前
深度学习---pytorch搭建深度学习模型(附带图片五分类实例)
pytorch·深度学习·分类
凢en22 分钟前
NOC科普一
网络·笔记·算法·智能路由器·硬件工程
yangmf20401 小时前
私有知识库 Coco AI 实战(四):打造 ES 索引参数小助手
大数据·人工智能·elasticsearch·coco ai
xixixiLucky1 小时前
爬虫学习笔记(一)
笔记·爬虫·学习