2019年8月3日
[深度学习] pytorch学习笔记（4）(Module类、实现Flatten类、Module类作用、数据增强)
摘要： 一、继承nn.Module类并自定义层 我们要利用pytorch提供的很多便利的方法，则需要将很多自定义操作封装成nn.Module类。 首先，简单实现一个Mylinear类： from torch import nn # Mylinear继承Module class Mylinear(nn.Mod 阅读全文
posted @ 2019-08-03 15:10 风间悠香 阅读(7380) 评论(2) 推荐(0)