Pytorch入门——两层神经网络(Numpy手写+Pytorch实现)

既然是实战,作为开篇,需要自己明确一下,理论的东西少整,实战就要多写代码。
先上一个重要的方法论吧,具有指导意义,也可用于学习其他技术上面:
如何成为PyTorch大神

1. Pytorch入门

简单总结一些常用的一些操作吧,只是一个非常浅显总结,还需要平时多加使用。


```1.tensor初始化```
x = torch.empty(5,3) #构造未初始化的tensor
x = torch.rand(5,3) #构造0~1的随机初始化矩阵
x = torch.randn(5,3) #构造均值为0,方差为1的随机初始化矩阵
x = torch.zeros(5,3,dtype=torch.long) #构造全为0,类型为long的矩阵
x = torch.tensor([1,2,3]) #从数据直接构建tensor
x = x.new_ones(5,3,dtype=torch.double) #从已有tensor构建,重用原来的tensor特征,如:数据类型。
x = torch.randn_like(x,dtype=torch.float) #将原来的tensor改造成均值为0,方差为1的新tensor

```2.tensor运算```
x.shape #返回tensor形状,返回一个tuple
a+b #两种加的实现
torch.add(a,b)
note:
 - 任何in-place的运算都以_结尾。如:x.copy_(y),x_t_(),会改变x。
 - y.add_(x)意思:把x加到y上,并替换y。
 - 
```3.类似numpy的操作(indexing、reshape、tensor与numpy的转换)```
x[1:,1:] #各种类似Numpy的indexing都可以在tensor上使用
x = torch.randn(4,4)
x.review(2,8) #2×8,类似numpy的reshape
x.review(-1,8) #按第2维,列数是8
x.review(2,-1) #按第1维,行数是2
x.item() #如果tensor只有一个元素,可以取出
b = torch.from_numpy(a)
```4.CUDA Tensors```
x.to() #tensor可以被移动到别的device上

2. 两层神经网络的实现

2.1 Numpy手写

import numpy as np
#超参数设置
N, x_dim, h_dim, y_dim &
  • 4
    点赞
  • 13
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
当使用PyTorch实现两层神经网络进行预测时,你需要进行以下步骤: 1. 导入必要的库和模块: ```python import torch import torch.nn as nn import torch.optim as optim ``` 2. 定义神经网络模型: ```python class TwoLayerNet(nn.Module): def __init__(self, input_size, hidden_size, output_size): super(TwoLayerNet, self).__init__() self.fc1 = nn.Linear(input_size, hidden_size) self.relu = nn.ReLU() self.fc2 = nn.Linear(hidden_size, output_size) def forward(self, x): out = self.fc1(x) out = self.relu(out) out = self.fc2(out) return out ``` 在这个例子中,我们定义了一个继承自`nn.Module`的`TwoLayerNet`类,其中包含两个全连接层和一个ReLU激活函数。 3. 准备数据: ```python # 假设你已经有了训练数据和标签数据 # 将数据转换为Tensor类型 x_train = torch.Tensor(train_data) y_train = torch.Tensor(train_labels) ``` 4. 实例化模型和定义损失函数与优化器: ```python input_size = 10 # 输入特征的维度 hidden_size = 20 # 隐藏层的大小 output_size = 1 # 输出的维度 model = TwoLayerNet(input_size, hidden_size, output_size) criterion = nn.MSELoss() # 均方误差损失函数 optimizer = optim.SGD(model.parameters(), lr=0.01) # 随机梯度下降优化器 ``` 5. 训练模型: ```python num_epochs = 100 # 迭代次数 for epoch in range(num_epochs): # 前向传播 outputs = model(x_train) loss = criterion(outputs, y_train) # 反向传播和优化 optimizer.zero_grad() loss.backward() optimizer.step() # 打印训练信息 if (epoch+1) % 10 == 0: print('Epoch [{}/{}], Loss: {:.4f}'.format(epoch+1, num_epochs, loss.item())) ``` 6. 进行预测: ```python # 假设你有测试数据 x_test = torch.Tensor(test_data) # 关闭梯度计算 with torch.no_grad(): predictions = model(x_test) # 打印预测结果 print(predictions) ```

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值