在Pytorch中,可以使用nn.Parameter实现此操作,示例如下:param = nn.Parameter(torch.Tensor(out_channels, inter_channels, 1, 1))nn.init.normal_(param, mean=0.001, std=0.001)
在Pytorch中,可以使用nn.Parameter实现此操作,示例如下:param = nn.Parameter(torch.Tensor(out_channels, inter_channels, 1, 1))nn.init.normal_(param, mean=0.001, std=0.001)