用pytorch的两种方法创建神经网络
2023-09-14 09:05:37 时间
import torch
import torch.nn.functional as F
replace following class code with an easy sequential network
class Net(torch.nn.Module):
def init(self, n_feature, n_hidden, n_output):
super(Net, self).init()
self.hidden = torch.nn.Linear(n_feature, n_hidden) # hidden layer
self.predict = torch.nn.Linear(n_hidden, n_output) # output layer
def forward(self, x):
x = F.relu(self.hidden(x)) # activation function for hidden layer
x = self.predict(x) # linear output
return x
net1 = Net(1, 10, 1)
easy and fast way to build your network
net2 = torch.nn.Sequential(
torch.nn.Linear(1, 10),
torch.nn.ReLU(),
torch.nn.Linear(10, 1)
)
print(net1) # net1 arch
相关文章
- 图神经网络的可解释性方法介绍和GNNExplainer解释预测的代码示例(附代码)
- ESRGCNN:西工大&CityU&中南大学&NTHU&哈工大&CUHK联合提出一种增强组卷积神经网络的图像超分辨方法
- 方法调用机制分析
- 图神经网络的可解释性方法介绍和GNNExplainer解释预测的代码示例
- 实验设计(DOE)的正确使用方法
- 基于深度神经网络的「端到端」学习位移场的方法,用于粒子图像测速
- Linux下查询最大连接数的方法(linux查看最大连接数)
- 提高MSSQL查询效率的有效方法(mssql查询效率)
- 时间设置处理Java中Redis Key过期时间的方法(redisjava过期)
- 一种基于神经网络的前所未有的方法可模拟量子系统
- 图像处理一定要用卷积神经网络?这里有一个另辟蹊径的方法
- Linux下快速简便创建群组的方法(linux创建群组)
- Redis封装解决瓶颈问题的唯一方法(先查redis封装)
- Linux系统下常见基本问题的解决方法
- CHECKBOX的全选、取消及跨页保存的实现方法
- 计算一个Java对象占用字节数的方法
- 原生js实现复制对象、扩展对象类似jquery中的extend()方法
- JS的location.href跳出框架打开新页面的方法