时间戳

首页 > 解决方案 > Pytorch 和张量在每组数字内进行操作

python - Pytorch 和张量在每组数字内进行操作

问题描述

我有以下 Pytorch 张量：

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 的情况下做到这一点。

有没有办法做到这一点？

标签： pythonsumpytorchtensor

解决方案

你可以做

torch.abs(V1[:, 1]- V1[:, 0])

并总结一下

torch.sum(torch.abs(V1[:, 1]- V1[:, 0]))

推荐阅读