分类目录：《深入浅出PaddlePaddle函数》总目录
相关文章：
· 深入浅出PaddlePaddle函数——paddle.Tensor
· 深入浅出PaddlePaddle函数——paddle.ones
· 深入浅出PaddlePaddle函数——paddle.zeros
· 深入浅出PaddlePaddle函数——paddle.full
· 深入浅出PaddlePaddle函数——paddle.ones_like
· 深入浅出PaddlePaddle函数——paddle.zeros_like
· 深入浅出PaddlePaddle函数——paddle.full_like

创建一个形状为shape、数据类型为dtype且值全为1的Tensor。

语法

paddle.ones(shape, dtype=None, name=None)

参数

  • shape：[tuple/list/Tensor] 要创建的Tensor的形状，shape的数据类型为int32或int64。
  • dtype：[可选，np.dtype/str] 要创建的Tensor的数据类型，可以为bool、float16、float32、float64、int32 或int64。如果dtype为None，那么数据类型为float32。
  • name：[可选，str] 具体用法请参见Name，一般无需设置，默认值为None。

返回值

Tensor，每个元素都是1，形状为 shape，数据类型为dtype。

实例

import paddle

# default dtype for ones OP
data1 = paddle.ones(shape=[3, 2])
# [[1. 1.]
#  [1. 1.]
#  [1. 1.]]

data2 = paddle.ones(shape=[2, 2], dtype='int32')
# [[1 1]
#  [1 1]]

函数实现

def ones(shape, dtype=None, name=None):
    """
    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 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 are 1.
    Examples:
        .. code-block:: python
            import paddle
            # default dtype for ones OP
            data1 = paddle.ones(shape=[3, 2])
            # [[1. 1.]
            #  [1. 1.]
            #  [1. 1.]]
            data2 = paddle.ones(shape=[2, 2], dtype='int32')
            # [[1 1]
            #  [1 1]]
            # shape is a Tensor
            shape = paddle.full(shape=[2], dtype='int32', fill_value=2)
            data3 = paddle.ones(shape=shape, dtype='int32')
            # [[1 1]
            #  [1 1]]
    """
    if dtype is None:
        dtype = 'float32'
    return fill_constant(value=1.0, shape=shape, dtype=dtype, name=name)

更多推荐

深入浅出PaddlePaddle函数——paddle.ones