摘要: 官网解释 一个顺序容器。每个模块将被按顺序添加到其中。Sequential 的 forward() 方法接受任何输入并将其转发到它的第一个模块。之后第一个模块输出的值将被第二个模块接收作为输入,以此类推直到最后一个模块的输出。 与手动调用一系列模块相比,Sequential 提供的价值在于它允许将整 阅读全文
posted @ 2022-08-12 22:34 Zongxi_giegie 阅读(570) 评论(0) 推荐(0)
摘要: 本文主要是将官网的解释翻译一下 官网解释: torch.nn.Flatten(start_dim=1, end_dim=- 1) 将一个张量在连续维度内进行展平,用于Sequential 输入和输出形状: 输入形状: $(,S_{start},...,S_i,...,S_{end},)$ 输出形状: 阅读全文
posted @ 2022-08-12 11:26 Zongxi_giegie 阅读(729) 评论(0) 推荐(0)