PyTorch神经网络模块

作者Frank

torch.nn

torch.nn是PyTorch的一个class
包含Module、Parameter和functional三个模块
Module：创建一个可调用对象。它的行为类似于函数，但同时也能包含状态（例如神经网络层的权重）。它知晓自己内部包含哪些Parameter，并具备梯度归零、便利参数以进行权重更新等功能

类似文章

发表回复