PyTorch - torch.ones_like、torch.zeros_like、 torch.full_like

最新推荐文章于 2024-10-29 14:21:56 发布

二分掌柜的

最新推荐文章于 2024-10-29 14:21:56 发布

阅读量1.2k

点赞数 1

分类专栏： # PyTorch 文章标签： ones_like zeros_like full_like

本文链接：https://blog.csdn.net/flyfish1986/article/details/119059965

版权

PyTorch 专栏收录该内容

37 篇文章

订阅专栏

本文介绍了PyTorch中创建特定数值的张量方法，包括ones_like、zeros_like及full_like函数的应用。通过实例展示了如何根据已有张量的形状创建全1、全0或指定数值填充的新张量。

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

PyTorch - torch.ones_like、 torch.zeros_like、 torch.full_like

flyfish

import torch
input = torch.rand(3, 4)
print(input)
# tensor([[0.5840, 0.8260, 0.7539, 0.2138],
#         [0.9743, 0.0964, 0.7610, 0.5746],
#         [0.6247, 0.3334, 0.6949, 0.9065]])

# 与input形状相同、元素全为1
a = torch.ones_like(input)
print(a)
# tensor([[1., 1., 1., 1.],
#         [1., 1., 1., 1.],
#         [1., 1., 1., 1.]])

#与input形状相同、元素全为0
b = torch.zeros_like(input)
print(b)
# tensor([[0., 0., 0., 0.],
#         [0., 0., 0., 0.],
#         [0., 0., 0., 0.]])

#与input形状相同、元素全为3
c = torch.full_like(input,3)
print(c)
# tensor([[3., 3., 3., 3.],
#         [3., 3., 3., 3.],
#         [3., 3., 3., 3.]])

print(torch.zeros(3,2))
# tensor([[0., 0.],
#         [0., 0.],
#         [0., 0.]])

print(torch.ones(3,2))
# tensor([[1., 1.],
#         [1., 1.],
#         [1., 1.]])