torch.nn
是专门为神经网络设计的模块化接口,nn构建于autgrad
之上,可以用来定义和运行神经网络。torch.nn.Module
是nn中重要的类,包含网络各层的定义,以及forward方法。这个类的内部有多达 48 个函数,这个类是 PyTorch 中所有 neural network module 的基类,自己创建的网络模型都是这个类的子类。
PyTorch | torch.nn.Module
最新推荐文章于 2022-11-20 15:42:55 发布
torch.nn
是专门为神经网络设计的模块化接口,nn构建于autgrad
之上,可以用来定义和运行神经网络。torch.nn.Module
是nn中重要的类,包含网络各层的定义,以及forward方法。这个类的内部有多达 48 个函数,这个类是 PyTorch 中所有 neural network module 的基类,自己创建的网络模型都是这个类的子类。