自定义Module

PyTorch 是一个基于 python 的深度学习库。PyTorch 源码库的抽象层次少，结构清晰，代码量适…