2024年5月26日
nn.Linear输入输出维度的理解
摘要： 目录 最近写代码发现nn.Linear的输入维度其实可以不是一维的输入。 一般来说，nn.Linear的输入一般是(batch_size, input_dimen)。但是官方文档说明了： input:(，input_dimen) output:(，output_dimen) 因此其实*是可以任意维度 阅读全文
posted @ 2024-05-26 11:12 JaxonYe 阅读(226) 评论(0) 推荐(0)