本文主要是介绍Pytorch_linear,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!
Linear
对输入数据应用线性变换:y = xA^T + b
torch.nn.Linear(in_features, out_features, bias=True)
参数
in_features
每个输入样本的大小out_features
每个输出样本的大小bias
若为False
,layer不会学习附加偏差b
shape
-
输入:
(N, ∗, H_in)
,其中∗
代表任意数量的附加维度,H_in = in_features
-
输出:
(N, *, H_out)
,除了最后一个维度,其余都与输入相同,H_out = out_features
这篇关于Pytorch_linear的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!