torch.nn.Parameter是继承自torch.Tensor的子类,其主要作用是作为nn.Module中的可训练参数使用。
例如:
self.weight=nn.Parameter(torch.randn(ndim,1))
self.bias=nn.Parameter(torch.randn(1))
torch.nn.Parameter是继承自torch.Tensor的子类,其主要作用是作为nn.Module中的可训练参数使用。
例如:
self.weight=nn.Parameter(torch.randn(ndim,1))
self.bias=nn.Parameter(torch.randn(1))