pytorch教程之nn.Module类详解——使用Module类来自定义模型-CSDN博客

网站介绍:文章浏览阅读10w+次,点赞700次,收藏2.2k次。前言:pytorch中对于一般的序列模型,直接使用torch.nn.Sequential类及可以实现,这点类似于keras,但是更多的时候面对复杂的模型,比如:多输入多输出、多分支模型、跨层连接模型、带有自定义层的模型等,就需要自己来定义一个模型了。本文将详细说明如何让使用Mudule类来自定义一个模型。一、torch.nn.Module类概述个人理解,pytorch不..._nn.module