From ff1789ca5eba03fe47764cdfec7791f6d149eea4 Mon Sep 17 00:00:00 2001 From: BrilliantYuKaimin <91609464+BrilliantYuKaimin@users.noreply.github.com> Date: Wed, 1 Jun 2022 15:19:10 +0800 Subject: [PATCH] Update creation.py (#42915) --- python/paddle/tensor/creation.py | 13 +++++++------ 1 file changed, 7 insertions(+), 6 deletions(-) diff --git a/python/paddle/tensor/creation.py b/python/paddle/tensor/creation.py index e37ca981f85..5a1aa5dcfce 100644 --- a/python/paddle/tensor/creation.py +++ b/python/paddle/tensor/creation.py @@ -482,19 +482,20 @@ def full_like(x, fill_value, dtype=None, name=None): def ones(shape, dtype=None, name=None): """ - The OP creates a tensor of specified :attr:`shape` and :attr:`dtype`, and fills it with 1. + Create a Tensor of specified :attr:`shape` and :attr:`dtype` and fill it with 1. Args: - shape(tuple|list|Tensor): Shape of the Tensor to be created, the data type of shape is int32 or int64. - dtype(np.dtype|str, optional): Data type of output Tensor, it supports - bool, float16, float32, float64, int32 and int64. Default: if None, the data type is 'float32'. - name(str, optional): The default value is None. Normally there is no need for user to set this property. For more information, please refer to :ref:`api_guide_Name` + shape (tuple|list|Tensor): Shape of the Tensor to be created, the data type of shape should be int32 or int64. + dtype (np.dtype|str, optional): Data type of output Tensor, it should be one of + bool, float16, float32, float64, int32 and int64. If it is set to None, the data type will be float32. + name (str, optional): For details, please refer to :ref:`api_guide_Name`. Generally, no setting is required. Default: None. Returns: - Tensor: A tensor of data type :attr:`dtype` with shape :attr:`shape` and all elements set to 1. + Tensor: A Tensor of data type :attr:`dtype` with shape :attr:`shape` and all elements are 1. Examples: .. code-block:: python + :name: ones-example import paddle -- GitLab