次のようなネットワークにある、1つ目のLinearのbiasの初期値を[1, 0, 0, 1, 0]に設定したいのですが、どうすれば良いのでしょうか?
import torch import torch.nn as nn class Net(nn.Module): def __init__(self): super(Net, self).__init__() self.main = nn.Sequential( nn.Linear(3, 5), nn.ReLU(), nn.Linear(5, 10), nn.ReLU()) def forward(self, input_data): out = self.main(input_data) return out net = Net() input_data = torch.randn(64, 3) output = net(input_data) '
よろしくお願い申し上げます。
回答3件
あなたの回答
tips
プレビュー
バッドをするには、ログインかつ
こちらの条件を満たす必要があります。