torch/nn目录结构以及__init__.py

torch/nn目录结构以及init.py

torch/nn目录结构

__init__.py:

from .modules import *
#nn.modules      导入modules目录下内容  定义容器modules
from .parameter import Parameter
#nn.Parameter    导入parameter.py     定义parameter
from .parallel import DataParallel
#导入parallel目录下data_parallel.py中的DataParallel类
from . im