parameters_to_vector¶

paddle.nn.utils. parameters_to_vector ( parameters, name=None ) [源代码] ¶

将输入的多个 parameter 展平并连接为 1 个 1-D Tensor。

参数¶

parameters (Iterable[Tensor]) - 可迭代的多个 parameter。parameter 为 Layer 中可训练的 Tensor。

name (str，可选) - 具体用法请参见 Name，一般无需设置，默认值为 None。

返回¶

Tensor，多个 parameter 展平并连接的 1-D Tensor

代码示例¶

          >>> import paddle
>>> paddle.seed(2023)
>>> linear = paddle.nn.Linear(10, 15)

>>> t = paddle.nn.utils.parameters_to_vector(linear.parameters())
>>> print(t.shape)
[165]

         