torch.zeros函数
时间: 2023-12-13 10:31:46 浏览: 97
PyTorch Zero To All
`torch.zeros()`函数是PyTorch中的一个函数,用于创建一个指定大小的全零张量(tensor)。该函数的语法如下:
```python
torch.zeros(*sizes, out=None, dtype=None, layout=torch.strided, device=None, requires_grad=False)
```
其中,`*sizes`表示张量的大小,可以是一个整数,也可以是一个元组或列表。`out`参数是一个可选的输出张量,`dtype`参数指定输出张量的数据类型,`layout`参数指定张量的布局,`device`参数指定张量所在的设备,`requires_grad`参数指定是否需要计算梯度。
以下是一个创建大小为(2,3)的全零张量的例子:
```python
import torch
x = torch.zeros(2, 3)
print(x)
```
输出结果为:
```
tensor([[0., 0., 0.],
[0., 0., 0.]])
```
阅读全文