© 1999-2048 dssz.net 粤ICP备11031372号
[其它] 用pytorch的nn.Module构造简单全链接层实例
说明:python版本3.7,用的是虚拟环境安装的pytorch,这样随便折腾,不怕影响其他的python框架 1、先定义一个类Linear,继承nn.Module import torch as t from torch import nn from torch.autograd import Variable as V class Linear(nn.Module): '''因为Variable自动求导,所以不需要实现backward()''' def __init__(self, i<weixin_38737635> 上传 | 大小:41kb