column_stack

paddle. column_stack ( x, name=None )

沿水平轴堆叠输入 x 中的所有张量。 x 中的每个张量 tensor 如果在堆叠之前 tensor.ndim < 2 ,则首先被整形为 (tensor.numel(), 1)。所有张量必须具有相同的数据类型。

下图展示了一个 column_stack 的情形,将一个形状为(3,1)的二维张量与一个形状为(3,)的一维张量进行 column_stack 操作,其中一维张量首先被整形为(3,1),然后再沿水平轴堆叠输入所有张量。通过图例能很好看出张量的变化。

图例

参数

  • x (list[Tensor]|tuple[Tensor]) - 输入 x 可以是张量的 list 或 tuple, x 中张量的数据类型必须相同。支持的数据类型: float16float32float64int32int64bfloat16

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

返回

Tensor,与输入数据类型相同的堆叠张量。

代码示例

COPY-FROM: paddle.column_stack