PyTorch02-神经网络的“组件” | 少年歌行

[lyb92nlf]

https://lyb92nlf.github.io/2022/01/18/pytorch02-%E7%A5%9E%E7%BB%8F%E7%BD%91%E7%BB%9C%E7%9A%84%E2%80%9C%E7%BB%84%E4%BB%B6%E2%80%9D/

torch.nn文档在这里！！！ ContainersMoudle对于所有神经网络的基础类。自定义模型需要继承这个类，必须重写forward方法 12345678910111213141516import torch.nn as nnclass Model(nn.Module): def init(self): super(Model, self).init()