ones

参数

• shape (list|tuple|Tensor|*shape) - 生成的 Tensor 的形状。数据类型为 int32 或 int64。 如果 shape 是 list、tuple，则其中的元素可以是 int，或者是形状为 [] 且数据类型为 int32、int64 的 0-D Tensor。 如果 shape 是 Tensor，则是数据类型为 int32、int64 的 1-D Tensor，表示一个列表。 如果 shape 是 *shape，则可以直接以可变参数的形式传入多个整数（例如 randn(2, 3)）。 该参数的别名为 size。

• dtype (str|paddle.dtype|np.dtype，可选) - 要创建的 Tensor 的数据类型，可以为 bool、float16、float32、float64、int32 或 int64。如果 dtype 为 None，那么数据类型为 float32。

• name (str，可选) - 具体用法请参见 api_guide_Name，一般无需设置，默认值为 None。

关键字参数

• out (Tensor，可选) - 用于存储结果的 Tensor。若指定，将直接写入该 Tensor，默认值为 None。

• device (PlaceLike|None，可选) - 期望创建 Tensor 所在的设备。默认值为 None，表示使用当前全局设备（可通过 paddle.device.set_device 设置）。

• requires_grad (bool，可选) - 是否需要为返回的 Tensor 记录梯度信息。默认值为 False。

• pin_memory (bool，可选) - 若为 True，返回的 CPU Tensor 将分配在锁页内存中。仅对 CPU Tensor 生效。默认值为 False。

返回

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

代码示例

>>> import paddle

>>> # shape is a list/tuple
>>> data1 = paddle.ones(shape=[3, 2])
>>> print(data1.numpy())
[[1. 1.]
 [1. 1.]
 [1. 1.]]

>>> # shape is a Tensor
>>> shape = paddle.to_tensor([3, 2])
>>> data2 = paddle.ones(shape=shape)
>>> print(data2.numpy())
[[1. 1.]
 [1. 1.]
 [1. 1.]]

>>> # shape is a Tensor List
>>> shape = [paddle.to_tensor(3), paddle.to_tensor(2)]
>>> data3 = paddle.ones(shape=shape)
>>> print(data3.numpy())
[[1. 1.]
 [1. 1.]
 [1. 1.]]

使用本API的教程文档¶

• 报错调试
• 隐式数据类型提升介绍
• CV - 迁移经验汇总
• NLP - 迁移经验汇总
• PyTorch 写法
• PyTorch 写法
• PyTorch 写法
• PyTorch 写法
• 迁移经验汇总
• Tensor 介绍
• Paddle Tensor 索引介绍