1.工作机制:
- net.parameters()返回了一个生成器,生成的是模型中所有可以训练的参数
- 每次循环,遍历了某一层的权重或者偏置
net.parameters() 输出为:[fc1.weight, fc1.bias, fc2.weight, fc2.bias] for param in net.parameters(): print(param.shape) torch.Size([5, 10]) # fc1.weight torch.Size([5]) # fc1.bias torch.Size([2, 5]) # fc2.weight torch.Size([2]) # fc2.bias
从上面的代码可以看出,每循环一次,输出某一层的权重或者偏置,一直到把所有层都输出完毕