为什么nn.Linear 的weight 是 (out_features, in_features)
xiezhengcai
·
2025-03-26
·
via 博客园 - xiezhengcai
在PyTorch的nn.Linear中,权重矩阵的形状为(out_features, in_features)。这是因为线性变换的实现方式为: 具体来说: 当创建nn.Linear(10, 60)时,in_featur…
此内容由惯性聚合(RSS阅读器)自动聚合整理,仅供阅读参考。 原文来自 — 版权归原作者所有。