Torch.nn.Module 模块官方文档学习
1、nn.parameterTensor类的特殊化实例,利用torch.nn.Parameter(data,required_grad = True)定义的变量可以直接作为网络模型的parameter2、nn.Moduletorch中搭建神经网络的父类add_module(name,module)向当前模块中添加子模块,添加的模块可以通过name访问apply(fn)...
原创
2019-08-18 18:15:20 ·
1322 阅读 ·
0 评论