首页
学习
活动
专区
圈层
工具
发布
社区首页 >专栏 >torch.nn.Linear(in_features, out_features, bias=True)[source]

torch.nn.Linear(in_features, out_features, bias=True)[source]

作者头像
狼啸风云
发布于 2022-08-18 14:23:11
发布于 2022-08-18 14:23:11
7350
举报

对到来的数据使用线性变换:

参数:

  • in_features – 每个输入样本的大小
  • out_features – 每个输出样本的大小
  • bias – 如果设置为False,层将不会学习添加偏差。默认值:True

形状:

  • 输入:

其中∗*∗ 的含义为附加维的任何数值,并且

  • 输出:

, 除了最后一个维度外,其他所有维度都与输入相同,并且

.

变量:

  • ~Linear.weight – shape (out_features,in_features)模块的可学习权值。从

得到的 初始化的值,其中

  • ~Linear.bias – 形状(out_features)模块的可学习偏差。 如果bias为真,则初始化值

,其中

例:

代码语言:javascript
复制
>>> m = nn.Linear(20, 30)
>>> input = torch.randn(128, 20)
>>> output = m(input)
>>> print(output.size())
torch.Size([128, 30])
本文参与 腾讯云自媒体同步曝光计划,分享自作者个人站点/博客。
原始发表:2022-08-17,如有侵权请联系 cloudcommunity@tencent.com 删除
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档