ToB企服应用市场:ToB评测及商务社交产业平台

标题: pytorch底子-nn.linear [打印本页]

---

作者: 耶耶耶耶耶    时间: 3 天前
标题: pytorch底子-nn.linear

1. import torch
2. import torch.nn as nn
4. # 定义线性层
5. linear_layer = nn.Linear(in_features=10, out_features=5, bias=True)
7. # 输入数据
8. input_data = torch.randn(32, 10)  # (batch_size=32, in_features=10)
10. # 前向传播
11. output = linear_layer(input_data)
12. print(output.shape)  # 输出形状: (32, 5)

复制代码

 维度变革

• 输入：(batch_size, in_features)。
  
• 输出：(batch_size, out_features)。
  

示例

• 输入形状：(32, 10)。
  
• 线性层：nn.Linear(10, 5)。
  
• 输出形状：(32, 5)。
  

实现细节：矩阵乘法

免责声明：如果侵犯了您的权益，请联系站长，我们会及时删除侵权内容，谢谢合作！更多信息从访问主页：qidao123.com:ToB企服之家，中国第一个企服评测及商务社交产业平台。

---

欢迎光临 ToB企服应用市场:ToB评测及商务社交产业平台 (https://dis.qidao123.com/)

Powered by Discuz! X3.4