Rate this Page

torch.nn.utils.parameters_to_vector#

torch.nn.utils.parameters_to_vector(parameters)[source]#

Flatten an iterable of parameters into a single vector.

Parameters

parameters (Iterable[Tensor]) – an iterable of Tensors that are the parameters of a model.

Returns

The parameters represented by a single vector

Return type

Tensor