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() 函数常用于初始化模型参数、创建占位符张量等场景。

Logo

旨在为数千万中国开发者提供一个无缝且高效的云端环境,以支持学习、使用和贡献开源项目。

更多推荐