快捷方式

torch.nn.utils.parameters_to_vector

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

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

参数

parameters (Iterable[Tensor]) – 一个包含模型参数的 Tensor 可迭代对象。

返回

表示参数的单个向量

返回类型

Tensor


© 版权所有 PyTorch 贡献者。

使用由 Read the Docs 提供的 Sphinx 主题构建。

文档

访问 PyTorch 的全面开发者文档

查看文档

教程

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

查看教程

资源

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

查看资源