如何在PyTorch中给出单个数字参数?

Question

我想用一个数字作为参数。

我试图插入如下参数

def __init__(self, ...)中的self.p1 = nn.Parameter()def _initialize(self):中的self.p1 = 1.0

但是我得到了错误

raise TypeError("cannot assign '{}' as parameter '{}' "
TypeError: cannot assign 'float' as parameter 'p1' (torch.nn.Parameter or None expected)

我如何使用一个单一的数字作为参数在我的模型初始化与1?

Answer 1

传递给nn.Parameter的参数应该是torch.Tensor:

>>> self.p1 = nn.Parameter(torch.tensor([1.0]))