nn.Parameter(函数作用)
时间: 2023-05-20 09:06:44 浏览: 79
PyTorch里面的torch.nn.Parameter()详解
3星 · 编辑精心推荐
nn.Parameter 是 PyTorch 中的一个类,用于将一个 Tensor 转换为可训练的参数。在神经网络中,我们通常需要训练一些参数,如权重和偏置,以使模型能够更好地拟合数据。使用 nn.Parameter 可以将这些参数标记为可训练的,并且在反向传播时会自动计算梯度。
阅读全文
PyTorch里面的torch.nn.Parameter()详解