From 2d937b79664d145181edcb74a8940f2a7f99e107 Mon Sep 17 00:00:00 2001 From: sandyhouse Date: Thu, 20 Aug 2020 15:36:44 +0000 Subject: [PATCH] update --- doc/fluid/api_cn/tensor_cn/expand_cn.rst | 32 ++++++++++++++++++++++-- 1 file changed, 30 insertions(+), 2 deletions(-) diff --git a/doc/fluid/api_cn/tensor_cn/expand_cn.rst b/doc/fluid/api_cn/tensor_cn/expand_cn.rst index 5a9ae8cba..2081fb08e 100644 --- a/doc/fluid/api_cn/tensor_cn/expand_cn.rst +++ b/doc/fluid/api_cn/tensor_cn/expand_cn.rst @@ -1,7 +1,35 @@ -.. _cn_api_tensor_cn_expand: +.. _cn_api_tensor_expand: expand ------------------------------- -:doc_source: paddle.fluid.layers.expand +.. 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 + 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]] -- GitLab