PyTorch 在torch.nn https://docs.pytorch.org/docs/stable/nn.html#module-torch.nn下提供 module,包括 常见层(卷积、池化、RNN、Transformer)、激活函数、损失函数等。
torch.nn
→ Abstract of VGON