小编Dri*_*iss的帖子

在 Pytorch 中创建具有固定权重的线性模型

我想在 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)

在此输入图像描述

这是可以实现的吗?如果是这样,我该怎么办?或者有替代的线性函数吗?

python neural-network pytorch

2
推荐指数
1
解决办法
2228
查看次数

标签 统计

neural-network ×1

python ×1

pytorch ×1