小编use*_*043的帖子

两个张量之间的 Pytorch 差异

我有两个这样的张量:

1st tensor
[[0,0],[0,1],[0,2],[1,3],[1,4],[2,1],[2,4]]

2nd tensor
[[0,1],[0,2],[1,4],[2,4]]
Run Code Online (Sandbox Code Playgroud)

我希望结果张量是这样的:

[[0,0],[1,3],[2,1]] # differences between 1st tensor and 2nd tensor
Run Code Online (Sandbox Code Playgroud)

我尝试使用 set、list、torch.where 等,但找不到任何好方法来实现这一点。有没有办法在两个不同大小的张量之间获得不同的行?(需要高效)

python pytorch tensor

4
推荐指数
1
解决办法
2132
查看次数

标签 统计

python ×1

pytorch ×1

tensor ×1