parameters_to_vector¶
将输入的多个 parameter 展平并连接为 1 个 1-D Tensor。
参数¶
parameters (Iterable[Tensor]) - 可迭代的多个 parameter。parameter 为 Layer 中可训练的 Tensor。
name (str,可选) - 具体用法请参见 Name,一般无需设置,默认值为 None。
返回¶
Tensor
,多个 parameter 展平并连接的 1-D Tensor
代码示例¶
import paddle
linear = paddle.nn.Linear(10, 15)
paddle.nn.utils.parameters_to_vector(linear.parameters())
# 1-D Tensor: [165]