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

Question

我想使用单个数字作为参数。

我试过如下插入参数

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

但是我收到了错误

    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]))

How to give single number parameter in PyTorch?