torch.zeros¶
- torch.zeros(*size, *, out=None, dtype=None, layout=torch.strided, device=None, requires_grad=False) Tensor ¶
返回一个填充了标量值 0 的张量,其形状由可变参数
size
定义。- 参数
size (int...) – 一个定义输出张量形状的整数序列。可以是可变数量的参数,也可以是列表或元组等集合。
- 关键字参数
out (Tensor, 可选) – 输出张量。
dtype (
torch.dtype
, 可选) – 返回张量所需的数据类型。默认值:如果None
,则使用全局默认值(参见torch.set_default_dtype()
)。layout (
torch.layout
, 可选) – 返回张量所需的布局。默认值:torch.strided
。device (
torch.device
, 可选) – 返回张量所需的设备。默认值:如果None
,则使用默认张量类型的当前设备(参见torch.set_default_device()
)。对于 CPU 张量类型,device
将是 CPU,对于 CUDA 张量类型,device
将是当前 CUDA 设备。requires_grad (bool, 可选) – 如果自动求导(autograd)应该记录对返回张量的操作。默认值:
False
。
示例
>>> torch.zeros(2, 3) tensor([[ 0., 0., 0.], [ 0., 0., 0.]]) >>> torch.zeros(5) tensor([ 0., 0., 0., 0., 0.])