暴躁早晨

文章
3
资源
0
加入时间
2年10月17天

pytorch的python API略读--torch.nn.Module

作者:机器视觉全栈er网站:www.cvtutorials.com2.3 torch.nn本节我们看下torch.nn,这个模块是基础模块。2.3.1 容器torch.nn.Module:这个类是所有网络模块的基类,甚至你写普通的数学函数的时候,也可以用这个作为基类。内置的模块和你自定义的模块都可以相互嵌套,构成一个树状的模块结构。下面我们来看下这个Module类常用案例:import torchimport torch.nn as nnimport torch.nn.functional