expand_cn.rst 1.2 KB
Newer Older
S
update  
sandyhouse 已提交
1
.. _cn_api_tensor_expand:
S
swtkiwi 已提交
2 3 4

expand
-------------------------------
S
swtkiwi 已提交
5

S
update  
sandyhouse 已提交
6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25
.. py:function:: paddle.expand(x, shape, name=None)

根据 ``shape`` 指定的形状扩展 ``x`` ,扩展后, ``x`` 的形状和 ``shape`` 指定的形状一致。

``x`` 的维数和 ``shape`` 的元素数应小于等于6,并且 ``shape`` 中的元素数应该大于等于 ``x`` 的维数。扩展的维度的维度值应该为1。

参数
:::::::::
    - x (Tensor) - 输入的Tensor,数据类型为:bool、float32、float64、int32或int64。
    - shape (tuple|list|Tensor) - 给定输入 ``x`` 扩展后的形状,若 ``shape`` 为list或者tuple,则其中的元素值应该为整数或者1-维Tensor,若 ``shape`` 类型为Tensor,则其应该为1-维Tensor。
    - name (str,可选) - 操作的名称(可选,默认值为None)。更多信息请参见 :ref:`api_guide_Name` 。

返回
:::::::::
``Tensor`` ,数据类型与 ``x`` 相同。

代码示例
:::::::::

.. code-block:: python
S
sandyhouse 已提交
26

S
update  
sandyhouse 已提交
27 28 29 30 31 32 33 34 35
       import paddle
       import numpy as np
               
       paddle.disable_static()
       np_data = np.array([1, 2, 3]).astype('int32')
       data = paddle.to_tensor(np_data)
       out = paddle.expand(data, [2, 3])
       np_out = out.numpy()
       # [[1, 2, 3], [1, 2, 3]]
S
swtkiwi 已提交
36