Python 我如何在第三维的中间创建一个全零的Py火炬张量和1？

我有一个张量，

torch.Size（[161，161，11]）

我想把它全部设置为零，我可以使用：

self.conv1.weight.data=torch.zeros（self.conv1.weight.data.Size（））

---

除此之外，我希望第三维度的第6列（中间）都是1。我该怎么做？

您可以在以后分配它：

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

或者，如果该张量是可训练的：

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

---

您可以在以后分配它：

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

或者，如果该张量是可训练的：

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