PyTorch学习笔记(六)——Sequential类、参数管理与GPU
目录一、torch.nn.Sequential一、torch.nn.SequentialSequential 本质是一个模块(即 Module),根据Pytorch中的约定,模块中可以继续添加模块。这意味着我们可以在 Sequential 中添加其它的模块(当然也可以添加其他的 Sequential)。添加完成后,Sequential 会将这些模块组成一个流水线,输入将依次通过这些模块得到一个输出,如下图所示:...
复制链接