快捷方式

torch.nn.utils.parameters_to_vector

torch.nn.utils.parameters_to_vector(parameters)[源代码]

将参数的可迭代对象展平为单个向量。

参数

parameters (Iterable[Tensor]) – 模型参数的可迭代张量。

返回值

由单个向量表示的参数

返回类型

Tensor

文档

访问 PyTorch 的全面开发者文档

查看文档

教程

获取面向初学者和高级开发人员的深入教程

查看教程

资源

查找开发资源并获得问题的解答

查看资源