今天遇到了这个函数
torch.nn.utils.parameters_to_vector()
其实很简单,它就是将一个多维tensor(tensor形状可以不规则,如网络模型参数的梯度)变为一个一维向量。
分割线上方就是输入torch.nn.utils.parameters_to_vector()的参数;
分割线下方就是torch.nn.utils.parameters_to_vector()输出的一维向量长度。
但是我不是很清楚下面这个函数是什么意思?
torch.nn.utils.vector_to_parameters()
看了一下有两个输入参数,作用是Convert one vector to the parameters。但是这个函数并没有返回值。