torch.zeros
时间: 2023-07-03 09:14:06 浏览: 78
torch.zeros() 函数是 PyTorch 中的一个函数,用于创建一个全零的张量(tensor)。该函数的语法如下:
```python
torch.zeros(*sizes, out=None, dtype=None, layout=torch.strided, device=None, requires_grad=False) -> Tensor
```
其中,参数:
- `sizes`:一个序列,表示要创建的张量的形状。每个元素代表每一维的长度。
- `out` (可选):输出张量。
- `dtype` (可选):输出张量的数据类型。
- `layout` (可选):输出张量的布局。
- `device` (可选):输出张量所在的设备。
- `requires_grad` (可选):是否需要计算梯度,默认为 False。
示例:
```python
import torch
# 创建一个形状为 (2, 3) 的全零张量
x = torch.zeros(2, 3)
print(x)
```
输出:
```
tensor([[0., 0., 0.],
[0., 0., 0.]])
```