torch.zeros() 函数介绍

torch.zeros() 函数是 PyTorch 中的一个函数，用于创建一个指定大小的全零张量（tensor）。

函数的语法如下：

sqltorch.zeros(*sizes, out=None, dtype=None, layout=torch.strided, device=None, requires_grad=False)

参数说明：

• *sizes：一个或多个整数，用于指定张量的形状（shape）。可以传递多个参数来创建一个多维张量。
• out：可选参数，用于指定输出张量的位置。
• dtype：可选参数，用于指定输出张量的数据类型。默认为 None，表示使用默认数据类型。
• layout：可选参数，用于指定张量的布局。默认为 torch.strided。
• device：可选参数，用于指定张量的设备（CPU 或 GPU）。
• requires_grad：可选参数，指示是否为张量启用梯度计算。默认为 False。

函数返回一个新的张量，其元素全部为零，并具有指定的形状、数据类型和设备。

以下是一些示例：

import torch

# 创建一个大小为 (3, 4) 的全零张量
x = torch.zeros(3, 4)
print(x)
# 输出:
# tensor([[0., 0., 0., 0.],
#         [0., 0., 0., 0.],
#         [0., 0., 0., 0.]])

# 创建一个大小为 (2, 3, 4) 的全零张量，数据类型为整数（torch.int32）
y = torch.zeros(2, 3, 4, dtype=torch.int32)
print(y)
# 输出:
# tensor([[[0, 0, 0, 0],
#          [0, 0, 0, 0],
#          [0, 0, 0, 0]],
#
#         [[0, 0, 0, 0],
#          [0, 0, 0, 0],
#          [0, 0, 0, 0]]], dtype=torch.int32)

torch.zeros() 函数常用于初始化模型参数、创建占位符张量等场景。