zongxi-giegie

摘要： torch.nn.Module 官网解释 所有神经网络模块的基类。 我们的模型也应该继承这个类。 模块还可以包含其他模块，允许将它们嵌套在树结构中。我们可以将子模块分配为常规属性： import torch.nn as nn import torch.nn.functional as F class 阅读全文