nn.Linear(d, num_units, bias=True)
是PyTorch中定义的一个全连接线性层。其中,d
是输入特征的数量,num_units
是输出特征的数量,而bias
参数决定是否在这个线性变换中添加一个偏置项。
设置bias=True
与bias=False
的区别如下:
数学表示:
其中, W W W是权重矩阵, x x x是输入, b b b是偏置项, y y y是输出。
参数数量:
d * num_units + num_units
。d * num_units
。模型表达能力:
初始化:
训练时间与复杂性:
通常,在大多数场景中,默认启用偏置是有利的,因为它增加了模型的表达能力,而额外的计算成本相对较小。但在某些特定的架构或应用中,可能会选择禁用偏置。