pytorch中的神经网络子模块——torch.nn.Linear
发布时间
阅读量：
阅读量
在PyTorch框架中，线性模块的实现方式为：于初始化函数中设定权重参数与偏置参数的初始值。
class Linear(Module):
__constants__ = ['bias', 'in_features', 'out_features']
def __init__(self, in_features, out_features, bias=True):
super(Linear, self).__init__()
self.in_features = in_features
self.out_features = out_features
self.weight = Parameter(torch.Tensor(out_features, in_features))
if bias:
self.bias = Parameter(torch.Tensor(out_features))
else:
self.register_parameter('bias', None)
self.reset_parameters()
def forward(self, i全部评论 (0)
还没有任何评论哟~