摘要: 简介 pytorch中其实一般没有特别明显的Layer和Module的区别,不管是自定义层、自定义块、自定义模型,都是通过继承Module类完成的。其实Sequential类也是继承自Module类的。 torcn.nn是专门为神经网络设计的模块化接口。构建于autograd之上,可以用来定义和运行 阅读全文
posted @ 2020-04-23 15:14 Skye_Zhao 阅读(4156) 评论(0) 推荐(1) 编辑