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
# 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]]