pytroch用自定义的tensor初始化nn.sequential中linear或者conv层的一种简单方法。
话不多说,上代码,上面写的很清楚。importtorch.nnasnnimporttorchnet=nn.Sequential(nn.Linear(1024,512),nn.ReLU(inplace=True),nn.Linear(512,256),nn.ReLU(inplace=True),nn.Linear(256,6),)net[4].weight.data=torch.zeros(6,25