Как я могу создать тензор PyTorch со всеми нулями и 1 в середине третьего измерения?

Question

У меня есть тензор, torch.Size([161, 161, 11]), и я хочу установить все это в нули, что я могу сделать с: self.conv1.weight.data = torch.zeros(self.conv1.weight.data.size())

Кроме того, я хочу столбец 6 (середина) третьего измерения быть всеми. Как мне это сделать?

Answer 1

Вы можете назначить его впоследствии:

self.conv1.weight.data[:, :, 6] = 1.0

Или, если этот тензор обучаем:

with torch.no_grad():
    self.conv1.weight.data[:, :, 6] = 1.0