unsqueeze_cn.rst 1.3 KB
Newer Older
1
.. _cn_api_paddle_tensor_unsqueeze
T
tianshuo78520a 已提交
2 3
unsqueeze
-------------------------------
4 5 6

.. py:function:: paddle.tensor.unsqueeze(input, axes, name=None)

S
swtkiwi 已提交
7 8 9 10 11 12
:alias_main: paddle.unsqueeze
:alias: paddle.unsqueeze,paddle.tensor.unsqueeze,paddle.tensor.manipulation.unsqueeze
:update_api: paddle.fluid.layers.unsqueeze



13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32
该OP向输入(input)的shape中一个或多个位置(axes)插入维度。

**参数**:
        - **input** (Variable)- 多维 Tensor,数据类型为 float32, float64, int8, int32,或 int64。
        - **axes** (int|list|tuple|Variable) - 表示要插入维度的位置。数据类型是 int32 。如果 axes 的类型是 list 或 tuple,它的元素可以是整数或者形状为[1]的 Tensor 。如果 axes 的类型是 Variable,则是1-D Tensor。
        - **name** (str,可选)- 一般无需设置。默认值: None。

**返回**:扩展维度后的多维Tensor

**返回类型**:Variable

**代码示例**:

.. code-block:: python

    import paddle
    import paddle.fluid as fluid
    import numpy as np
    with fluid.dygraph.guard():
        input_1 = np.random.random([5, 10]).astype("int32")
D
dyning 已提交
33
        # input is a variable which shape is [5, 10]
34 35 36 37
        input = fluid.dygraph.to_variable(input_1)

        output = paddle.unsqueeze(input, axes=[1])
        # output.shape [5, 1, 10]