成功搭建神经网络(三)

快速搭建

  无论是搭建回归类型的神经网络也好,分类类型的也罢,都是通过 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 实现快速搭建。

  运行截图:

 

 

posted @ 2021-11-26 13:27  Sunshine_y  阅读(23)  评论(0编辑  收藏  举报