摘要:
神经网络层的搭建主要是两种方法,一种是使用类(继承torch.nn.Moudle),一种是使用torch.nn.Sequential来快速搭建。 1)首先我们先加载数据: 2)两种方法的模板: 2.1: 类(class):这基本就是固定格式,init中定义每个神经层的神经元个数,和神经元层数,for 阅读全文
摘要:
本文将自己在pytorch学习中遇见的各种问题整理起来,并且持续更新。 1:torch.Tensor和torch.tensor的区别 开始使用torch.tensor和torch.Tensor的时候发现结果都是一样的。都能生成新的张量。但根源上是有差别的。 torch.Tensor()是Python 阅读全文