torch.nn.utils.vector_to_parameters
Copy slices of a vector into an iterable of parameters.
Parameters
vecTensor- A single 1D tensor. The vector will be sliced and copied into each parameter in order.
parametersIterable<Tensor>- Iterable of Tensors to fill with values from vec. Parameters are modified in-place.
Examples
// Get current parameters as vector
const vec = torch.nn.utils.parameters_to_vector(model.parameters());
// Modify vec...
const modified_vec = vec.add(0.1);
// Copy back to parameters
torch.nn.utils.vector_to_parameters(modified_vec, model.parameters());