2018年7月28日
Function与Module的差异与应用场景
摘要： Function与Module都可以对pytorch进行自定义拓展，使其满足网络的需求，但这两者还是有十分重要的不同： Function一般只定义一个操作，因为其无法保存参数，因此适用于激活函数、pooling等操作；Module是保存了参数，因此适合于定义一层，如线性层，卷积层，也适用于定义一个网 阅读全文
posted @ 2018-07-28 16:35 我们都是好孩子会前端 阅读(211) 评论(0) 推荐(0) 编辑