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()
函数常用于初始化模型参数、创建占位符张量等场景。
更多推荐
已为社区贡献4条内容
所有评论(0)