torch.nn.utils.parameters_to_vector

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

Преобразование параметров в один вектор

Параметры:

parameters (Iterable[Tensor]) – итератор тензоров, являющихся параметрами модели.

Возвращает:

Параметры, представленные одним вектором

Тип возвращаемого значения:

Tensor