unbind¶

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

Removes a tensor dimension, then split the input tensor into multiple sub-Tensors. :param input: The input variable which is an N-D Tensor, data type being float32, float64, int32 or int64. :type input: Variable :param axis: A scalar with type int32|int64 shape [1]. The dimension along which to unbind. If $axis < 0$, the

System Message: ERROR/3 (/usr/local/lib/python3.8/site-packages/paddle/fluid/layers/nn.py:docstring of paddle.fluid.layers.nn.unbind, line 5)

Unexpected indentation.

dimension to unbind along is $rank(input) + axis$. Default is 0.

System Message: WARNING/2 (/usr/local/lib/python3.8/site-packages/paddle/fluid/layers/nn.py:docstring of paddle.fluid.layers.nn.unbind, line 6)

Block quote ends without a blank line; unexpected unindent.

Returns

The list of segmented Tensor variables.

Return type

list(Variable)

Example

System Message: ERROR/3 (/usr/local/lib/python3.8/site-packages/paddle/fluid/layers/nn.py:docstring of paddle.fluid.layers.nn.unbind, line 13)

Error in “code-block” directive: maximum 1 argument(s) allowed, 63 supplied.

.. code-block:: python
    import paddle
    # input is a variable which shape is [3, 4, 5]
    input = paddle.fluid.data(
         name="input", shape=[3, 4, 5], dtype="float32")
    [x0, x1, x2] = paddle.tensor.unbind(input, axis=0)
    # x0.shape [4, 5]
    # x1.shape [4, 5]
    # x2.shape [4, 5]
    [x0, x1, x2, x3] = paddle.tensor.unbind(input, axis=1)
    # x0.shape [3, 5]
    # x1.shape [3, 5]
    # x2.shape [3, 5]
    # x3.shape [3, 5]