在每一组数字中执行操作的Py手电筒和张量
在每一组数字中执行操作的Py手电筒和张量
提问于 2019-05-04 17:52:10
我有以下毕火炬张量：
V1 = torch.tensor([[2, 4], [6, 4], [5, 3]])我想做每一对数字的差值之和(应用绝对值)，如下代码所示；
result.sum(abs(2-4), abs(6-4), abs(5-3))我可以使用for语句来实现这一点：
total = 0
for i in range(0,vector.size(0)):
total = total + torch.abs(vector.data[i][1] - vector.data[i][0])但我不想用for来做。
有办法吗？
回答 2
Stack Overflow用户
回答已采纳
发布于 2019-05-04 17:57:03
你能做到的
torch.abs(V1[:, 1]- V1[:, 0])
然后总结一下
torch.sum(torch.abs(V1[:, 1]- V1[:, 0]))
Stack Overflow用户
发布于 2019-05-05 00:37:07
您可以使用更通用的方法，如下面的代码段：
In [46]: torch.sum(torch.abs(V1[:, :-1] - V1[:, 1:]))
Out[46]: tensor(6)页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接：
https://stackoverflow.com/questions/55985336
复制相关文章
相似问题
腾讯云开发者
