ones¶
创建一个形状为 shape、数据类型为 dtype 且值全为 1 的 Tensor。
参数¶
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
# shape is a list/tuple
data1 = paddle.ones(shape=[3, 2])
# [[1. 1.]
# [1. 1.]
# [1. 1.]]
# shape is a Tensor
shape = paddle.to_tensor([3, 2])
data2 = paddle.ones(shape=shape)
# [[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)
# [[1. 1.]
# [1. 1.]
# [1. 1.]]