问题描述
我有两个这样的张量:
a = tensor([1,3,4,5]).cuda()
b = tensor([0,1,2,5,6]).cuda()
我希望结果张量像这样:
tensor([0,6])
第一张量和第二张量之间的差异
有没有简单的解决方案可以在cpu和GPU上运行,并且比np快得多?
解决方法
暂无找到可以解决该程序问题的有效方法,小编努力寻找整理中!
如果你已经找到好的解决方法,欢迎将解决方案带上本链接一起发送给小编。
小编邮箱:dio#foxmail.com (将#修改为@)