两个一维张量之间的差异

Question

我有两个这样的张量：

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

我希望结果张量像这样：

tensor([0,2,6]) 

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

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