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， complex64，complex128。

  • 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)。

也可以参考下方的图示来理解 unstack 是如何对张量进行变换的。图中分别展示了一个形状为 (3, 4) 的 Tensor 在 axis=1 和一个形状为 (2, 3, 3) 的 Tensor 在 axis=1 的 unstack 操作。

图例

代码示例

>>> 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]