nn.Parameter(torch.zeros(self.output_dim, 1))
时间: 2023-11-12 19:19:36 浏览: 100
Pytorch|神经网络工具箱nn
这是一个技术问题,它似乎与网络参数有关。在PyTorch中,nn.Parameter(torch.zeros(self.output_dim, 1))被用来创建一个存储在计算图中的可训练参数(trainable parameter)。这个参数是一个output_dim x 1的张量,所有元素都被初始化为0。
阅读全文