两个一维张量之间的差异

问题描述

我有两个这样的张量:

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

我希望结果张量像这样:

tensor([0,6]) 

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

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

解决方法

暂无找到可以解决该程序问题的有效方法,小编努力寻找整理中!

如果你已经找到好的解决方法,欢迎将解决方案带上本链接一起发送给小编。

小编邮箱:dio#foxmail.com (将#修改为@)