unbind¶

paddle. unbind ( input, axis=0 ) [source]

Removes a tensor dimension, then split the input tensor into multiple sub-Tensors.

Parameters

• input (Tensor) – The input variable which is an N-D Tensor, data type being float32, float64, int32 or int64.

• axis (int32|int64, optional) – A scalar with type int32|int64 shape [1]. The dimension along which to unbind. If $axis < 0$, the dimension to unbind along is $rank(input) + axis$. Default is 0.

Returns

The list of segmented Tensor variables.

Return type

list(Tensor)

Example

import paddle
import numpy as np
# input is a variable which shape is [3, 4, 5]
np_input = np.random.rand(3, 4, 5).astype('float32')
input = paddle.to_tensor(np_input)
[x0, x1, x2] = paddle.unbind(input, axis=0)
# x0.shape [4, 5]
# x1.shape [4, 5]
# x2.shape [4, 5]
[x0, x1, x2, x3] = paddle.unbind(input, axis=1)
# x0.shape [3, 5]
# x1.shape [3, 5]
# x2.shape [3, 5]
# x3.shape [3, 5]