深入浅出PaddlePaddle函数——paddle.ones-Toy模板网

这篇具有很好参考价值的文章主要介绍了深入浅出PaddlePaddle函数——paddle.ones。希望对大家有所帮助。如果存在错误或未考虑完全的地方，请大家不吝赐教，您也可以点击"举报违法"按钮提交疑问。

分类目录：《深入浅出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。文章来源地址https://www.toymoban.com/news/detail-450162.html

实例

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)