Pytorch torch.add() torch.add_() 用法

最新推荐文章于 2025-01-22 15:15:35 发布

幽影相随

最新推荐文章于 2025-01-22 15:15:35 发布

阅读量1.4w

点赞数 5

分类专栏： Python 文章标签： pytorch add add_ _

本文链接：https://blog.csdn.net/weixin_43820352/article/details/125995863

版权

Python 专栏收录该内容

7 篇文章

订阅专栏

这篇博客介绍了PyTorch中torch.add()函数用于张量相加的操作，包括格式相同的直接相加和不同格式时的扩容相加。还探讨了带下划线的in-place运算，即修改原变量。通过实例代码展示了torch.add()的用法，包括常数乘以张量的加法和张量间的加法，并对比了in-place操作前后张量的变化。

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

简介

torch.add()：对两个张量进行相加，若格式不同则以复制的方式进行扩容后再相加。
以 “下划线 _” 结尾的，均为in-place。
可以简单理解为：修改了对应变量中的数值。
Torch里面所有带 “下划线 ” 的操作，都是in-place的。
torch.add()：对两个张量进行相加，但格式需相同。

实例代码

x = torch.arange(1., 6.)
print(x)
print(torch.topk(x, 3))

a = torch.randn(4)
print(a)
b = torch.randn(4, 1)
print(b)
# alpha * b + a, 维度不够的地方自动扩容
print(torch.add(a, b, alpha=10))

print("p, q:")
p = torch.randn(4)
print(p)
q = torch.randn(4)
print(q)
p.add(q, alpha=10)
print(p)
p.add_(q, alpha=10)
print(p)

运行结果

tensor([1., 2., 3., 4., 5.])
torch.return_types.topk(
values=tensor([5., 4., 3.]),
indices=tensor([4, 3, 2]))
tensor([ 0.4467, -1.0670, -0.2640, -0.9347])
tensor([[ 1.4514],
        [-1.9113],
        [-0.3101],
        [-0.3147]])
tensor([[ 14.9611,  13.4474,  14.2505,  13.5797],
        [-18.6661, -20.1797, -19.3767, -20.0474],
        [ -2.6547,  -4.1683,  -3.3653,  -4.0360],
        [ -2.7007,  -4.2144,  -3.4114,  -4.0821]])
p, q:
tensor([ 1.6037,  1.4193, -0.4076,  1.3390])
tensor([0.2374, 1.1236, 0.8439, 0.7530])
tensor([ 1.6037,  1.4193, -0.4076,  1.3390])
tensor([ 3.9780, 12.6551,  8.0318,  8.8687])