成功搭建神经网络(三)
快速搭建
无论是搭建回归类型的神经网络也好,分类类型的也罢,都是通过 class Net(torch.nn.Module) 搭建,本文介绍另一种搭建方法,这种方法可以更快速的搭建net 。
#快速搭建
net = torch.nn.Sequential(
torch.nn.Linear(2, 10),
torch.nn.ReLU(),
torch.nn.Linear(10, 2),
)
print(net)
直接调用torch中的 Sequential 实现快速搭建。
运行截图: