上一节中我们介绍了怎样定义PyTorch的模型,其中给出的示例都是用torch.nn中的层来完成的。这种定义方式易于理解,在实际场景下不一定利于使用。当模型的深度非常大时候,使用Sequential定义模型结构需要向其中添加几百行代码,使用起来不甚方便。
torch.nn
Sequential