nn.Parameter的作用

torch.nn.Parameter是继承自torch.Tensor的子类,其主要作用是作为nn.Module中的可训练参数使用。

例如:

  self.weight=nn.Parameter(torch.randn(ndim,1))
  self.bias=nn.Parameter(torch.randn(1))

你可能感兴趣的:(python)