应用
torch.zeros(2, 3)
tensor([[ 0., 0., 0.],
[ 0., 0., 0.]])
API
torch.zeros(*size, out=None, dtype=None, layout=torch.strided, device=None, requires_grad=False) → Tensor
参数 | 描述 |
---|---|
size (int…) | |
out (Tensor, optional) | |
dtype (torch.dtype, optional) | |
layout (torch.layout, optional) | |
device (torch.device, optional) | |
requires_grad (bool, optional) |
参考:
https://pytorch.org/docs/stable/generated/torch.zeros.html#torch.zeros