self.hidden=torch.nn.Liner(1,20)
时间: 2024-04-21 10:23:43 浏览: 120
浅析PyTorch中nn.Linear的使用
这段代码创建了一个全连接层(fully connected layer)并其赋值给了self.hidden变量。具体来说,torch.nn.Linear(1, 20)创建了一个线性层,它的输入特征数量为1,输出特征数量为20。这意味着输入张量的形状应为(N, 1),其中N是输入样本的数量,输出张量的形状为(N, 20)。这个线性层可以将输入特征映射到20个输出特征。self.hidden变量将这个线性层保存下来,方便在模型的其他地方使用。请注意,正确的类名是Linear(而不是Liner)。
阅读全文