column_stack¶
- paddle. column_stack ( x, name=None ) ¶
沿水平轴堆叠输入 x
中的所有张量。 x
中的每个张量 tensor
如果在堆叠之前 tensor.ndim < 2
,则首先被整形为 (tensor.numel(), 1)
。所有张量必须具有相同的数据类型。
参数¶
x (list[Tensor]|tuple[Tensor]) - 输入
x
可以是张量的 list 或 tuple,x
中张量的数据类型必须相同。支持的数据类型:float16
、float32
、float64
、int32
、int64
或bfloat16
。name (str,可选) - 具体用法请参见 Name,一般无需设置,默认值为 None。
返回¶
Tensor,与输入数据类型相同的堆叠张量。
代码示例¶
COPY-FROM: paddle.column_stack