nn.Linear(in_features, out_features) — аффинное преобразование y = xA^T + b с обучаемыми весом и bias.
Разбор
- Входной последней размерностью должен быть
in_features.
- Bias можно отключить
bias=False.
- Инициализация весов задаётся в reset_parameters.
- Частый блок MLP вместе с активацией.
Итог
Linear — плотный слой: матрица весов плюс смещение.