Pytorch小知识汇总三：两个或多个tensor逐元素求和sum

xiangyong58

已于 2022-06-29 21:06:50 修改

阅读量7.1k

点赞数 3

CC 4.0 BY-SA版权

分类专栏： Machine & Deep Learning 文章标签： pytorch python

于 2022-06-19 13:20:47 首次发布

本文链接：https://blog.youkuaiyun.com/xiangyong58/article/details/125356886

Machine & Deep Learning 专栏收录该内容

78 篇文章 ¥9.90 ¥99.00

订阅专栏

超级会员免费看

本文介绍了如何在PyTorch中使用Python的sum函数对两个或多个Tensor进行逐元素求和。重点强调了需将Tensor构成列表形式以得到预期结果。

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

使用python自带的sum函数把两个或多个tensor逐元素求和。但一定要注意把这些tensor组成列表形式sum([a, b, c])，若直接用sum(a,b,c)并不是想要的正确结果。如下：

import numpy as np
import torch

x1 = torch.tensor([[1,2,3],[0,1,2]])
x2 = torch.tensor([[2,3,4],[2,4,1]])
x11 = np.array([[1,2,3], [0,1,2]])
x22 = np.array([[2,3,4],[2,4,1]])

print(x1+x2)
print(x11+x22)
>>>
tensor([[3, 5, 7],
        [2, 5, 3]])
[[3 5 7]
 [2 5 3]]


print(sum(x1,x2))
print(sum(x11,x22))
>>>
tensor([[3, 6, 9],
        [3, 7, 6]])
[[3 6 9]
 [3 7 6]]


print(sum([x1,x2]))
print(sum([x11,x22]))
>>>tensor([[3, 5, 7],
        [2, 5, 3]])
[[3 5 7]
 [2 5 3]]

print(torch.add(x1,x2))
>>>
tensor([[3, 5, 7],
        [2, 5, 3]])

仅两个tensor求和时：