🔥PyTorch的nn.Linear()详解🌟
发布时间：2025-03-26 19:26:50来源：
在深度学习框架PyTorch中，`nn.Linear()` 是一个非常基础且重要的模块，用于实现全连接层（Fully Connected Layer）。简单来说，它负责将输入数据通过权重矩阵和偏置向量进行线性变换，公式为：y = xA^T + b。
首先，你需要定义它的输入维度 `in_features` 和输出维度 `out_features`。例如：
```python
linear_layer = nn.Linear(in_features=128, out_features=64)
```
上述代码创建了一个全连接层，将128维的输入映射到64维的输出。
其次，别忘了初始化权重和偏置！PyTorch 默认会随机初始化这些参数，但你也可以手动设置。比如：
```python
torch.nn.init.xavier_uniform_(linear_layer.weight)
torch.nn.init.zeros_(linear_layer.bias)
```
最后，在前向传播时，只需传入张量即可：
```python
output = linear_layer(input_tensor)
```
掌握 `nn.Linear()` 后，你可以轻松构建复杂的神经网络模型啦！✨
免责声明：本答案或内容为用户上传，不代表本网观点。其原创性以及文中陈述文字和内容未经本站证实，对本文以及其中全部或者部分内容、文字的真实性、完整性、及时性本站不作任何保证或承诺，请读者仅作参考，并请自行核实相关内容。 如遇侵权请及时联系本站删除。
