torch.nn.Parameter用法
torch.nn.Parameter 是 PyTorch 中的一个类,主要用于将张量标记为模型中可学习的参数。它是 torch.Tensor 的子类,具有一个关键特性:如果一个 Parameter 被赋值给 nn.Module 的属性,那么它会自动被添加到模型的参数列表中,这使得它在调用 model.parameters() 时可以被方便地访问和优化。自动注册:当 Parameter 设置为 nn.Module 的属性时,它会自动成为模型的一部分,不需要手动将其添加到 parameters() 列表中。




