Pytorch 两个一维张量的差异

我有两个像这样的张量：

a = tensor([1,3,4,5]).cuda()
        
b = tensor([0,1,2,3,4,5,6]).cuda()

tensor([0,2,6]) 

我希望结果张量如下：

a = tensor([1,3,4,5]).cuda()
        
b = tensor([0,1,2,3,4,5,6]).cuda()

tensor([0,2,6]) 

第一张量和第二张量之间的差异

---

有没有一种简单的解决方案可以在CPU和GPU上运行，并且比np快得多，set？

这能回答您的问题吗？尽管可以像@trialNerror所建议的那样在GPU上实现，但我怀疑任何解决方案都会比简单的方法慢：

set（b.tolist（））-set（a.tolist（））

或

np.intersect1d

，因为处理张量需要开销。