我想在 PyTorch 下创建一个具有单层的线性网络,但我希望手动初始化权重并保持固定。
例如模型的权重值:
layer = nn.Linear(4, 1, bias=False)
weights = tensor([[ 0.6],
[0.25],
[ 0.1],
[0.05]], dtype=torch.float64)
Run Code Online (Sandbox Code Playgroud)
这是可以实现的吗?如果是这样,我该怎么办?或者有替代的线性函数吗?