torch.nn.utils.parameters_to_vector
Flatten an iterable of parameters into a single vector.
Parameters
parametersIterable<Tensor>- Iterable of Tensors to flatten
Returns
Tensor– A single 1D tensor containing all the parameters concatenatedExamples
const vec = torch.nn.utils.parameters_to_vector(model.parameters());
console.log(vec.shape); // [total_num_parameters]