Pytorch中torch.full()，torch.ones()和torch.zeros()函数解析

最新推荐文章于 2025-03-27 15:34:25 发布

cv_lhp

最新推荐文章于 2025-03-27 15:34:25 发布

阅读量1.3w

点赞数 12

分类专栏： Pytorch基础文章标签： pytorch 深度学习神经网络 python

本文链接：https://blog.csdn.net/flyingluohaipeng/article/details/125103847

版权

Pytorch基础专栏收录该内容

50 篇文章

订阅专栏

本文详细介绍了PyTorch中的torch.full()、torch.ones()和torch.zeros()函数，用于创建指定大小并填充特定值的张量。通过示例展示了如何使用这些函数创建二维和一维张量，并输出结果。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

一. torch.full()函数解析

1. 官网链接

torch.full()，如下图所示：

2. torch.full()函数解析

torch.full(size, fill_value, *, out=None, dtype=None, layout=torch.strided, device=None, requires_grad=False) → Tensor

返回创建size大小的维度，里面元素全部填充为fill_value

3.代码举例

输出形状为(2,3)的二维张量，里面元素全部填充为5

x = torch.full(size=(2,3),fill_value=5)
x

输出结果如下：
tensor([[5, 5, 5],
        [5, 5, 5]])

二. torch.ones()函数解析

1. 官网链接

torch.ones()，如下图所示：

2. torch.ones()函数解析

torch.ones(*size, *, out=None, dtype=None, layout=torch.strided, device=None, requires_grad=False) → Tensor

返回创建size大小的维度，里面元素全部填充为1

3. 代码举例

a = torch.ones(2, 3)
b = torch.ones(5)
c = torch.ones(size=(3,4))
a,b,c

输出结果如下：
(tensor([[1., 1., 1.],
         [1., 1., 1.]]),
 tensor([1., 1., 1., 1., 1.]),
 tensor([[1., 1., 1., 1.],
         [1., 1., 1., 1.],
         [1., 1., 1., 1.]]))

三. torch.zeros()函数解析

1.官网链接

torch.zeros()，如下图所示：

2. torch.zeros()函数解析

torch.zeros(*size, *, out=None, dtype=None, layout=torch.strided, device=None, requires_grad=False) → Tensor

返回创建size大小的维度，里面元素全部填充为0

3. 代码举例

a = torch.zeros(2, 3)
b = torch.zeros(5)
c = torch.zeros(size=(3,4))
a,b,c

输出结果如下：
(tensor([[0., 0., 0.],
         [0., 0., 0.]]),
 tensor([0., 0., 0., 0., 0.]),
 tensor([[0., 0., 0., 0.],
         [0., 0., 0., 0.],
         [0., 0., 0., 0.]]))