zeros¶
创建形状为 shape
、数据类型为 dtype
且值全为 0 的 Tensor。
参数¶
shape (tuple|list|Tensor) - 输出 Tensor 的形状,
shape
的数据类型为 int32 或者 int64。dtype (np.dtype|str,可选) - 输出 Tensor 的数据类型,数据类型必须为 bool、float16、float32、float64、int32 或 int64。若为 None,数据类型为 float32,默认为 None。
name (str,可选) - 具体用法请参见 Name,一般无需设置,默认值为 None。
返回¶
值全为 0 的 Tensor,数据类型和 dtype
定义的类型一致。
代码示例¶
import paddle
# shape is a list/tuple
data1 = paddle.zeros(shape=[3, 2])
# [[0. 0.]
# [0. 0.]
# [0. 0.]]
# shape is a Tensor
shape = paddle.to_tensor([3, 2])
data2 = paddle.zeros(shape=shape)
# [[0. 0.]
# [0. 0.]
# [0. 0.]]
# shape is a Tensor List
shape = [paddle.to_tensor(3), paddle.to_tensor(2)]
data3 = paddle.zeros(shape=shape)
# [[0. 0.]
# [0. 0.]
# [0. 0.]]