应用
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
版权声明:本文为claroja原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接和本声明。