unstack

paddle. unstack ( x, axis=0, num=None ) [源代码]

将单个 dim 为 D 的 Tensor 沿 axis 轴 unpack 为 num 个 dim 为 (D-1) 的 Tensor。

参数

  • x (Tensor) – 输入 x 为 dim > 0 的 Tensor,

支持的数据类型:float32,float64,int32,int64。

  • axis (int | 可选) – 输入 Tensor 进行 unpack 运算所在的轴,axis 的范围为:[-D, D)

如果 axis < 0,则 \(axis = axis + dim(x)\),axis 的默认值为 0。

  • num (int | 可选) - axis 轴的长度,一般无需设置,默认值为 None

返回

长度为 num 的 Tensor 列表,数据类型与输入 Tensor 相同,dim 为 (D-1)

代码示例

import paddle
x = paddle.ones(name='x', shape=[2, 3, 5], dtype='float32')  # create a tensor with shape=[2, 3, 5]
y = paddle.unstack(x, axis=1)  # unstack with second axis, which results 3 tensors with shape=[2, 5]