pytorch使用(三)torch.zeros用法

发布时间 2023-07-20 22:23:50作者: DuJunlong

torch.zeros用法

torch.zeros() 是 PyTorch 中用来创建全 0 张量的函数。用法为 torch.zeros(size, out=None, dtype=None, layout=torch.strided, device=None, requires_grad=False)。

其中,size 参数表示张量的形状(shape),可以是一个整数或者一个包含多个整数的 tuple。例如,torch.zeros(3, 4) 就会创建一个 3x4 的全 0 矩阵。

out 参数表示输出张量。如果指定了这个参数,那么函数会将结果存储在这个张量里,而不是新建一个张量。

dtype 参数表示张量的数据类型,可以是 torch.float32、torch.int64 等。

layout、device、requires_grad 分别表示张量的内存布局、存储设备、是否需要求导,一般不用设置。

使用示例 :

import torch
#创建一个 3x4 的全 0 矩阵
x = torch.zeros(3, 4)
print(x)
#创建一个 2x3x4 的全 0 张量
x = torch.zeros(2, 3, 4)
print(x)