zongxi-giegie

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

摘要： 官网解释 一个顺序容器。每个模块将被按顺序添加到其中。Sequential 的 forward() 方法接受任何输入并将其转发到它的第一个模块。之后第一个模块输出的值将被第二个模块接收作为输入，以此类推直到最后一个模块的输出。 与手动调用一系列模块相比，Sequential 提供的价值在于它允许将整 阅读全文

摘要： 本文主要是将官网的解释翻译一下 官网解释： torch.nn.Flatten(start_dim=1, end_dim=- 1) 将一个张量在连续维度内进行展平，用于Sequential 输入和输出形状： 输入形状： $(,S_{start},...,S_i,...,S_{end},)$ 输出形状： 阅读全文