YES~ 2022-07-28 21:27 采纳率: 50%
浏览 57
已结题

pytorch批量返回tensor特定值索引

问题遇到的现象和发生背景

在pytorch,pyg中,
想从1万多个边(all_edges:tensor shape[2,10640])找到指定64个边(target_nodes:tensor shape[2,64])对应的索引,代码怎么写会快一点?

问题相关代码,请勿粘贴截图

idx = []
for u, v in target_edges:
idx.append(all_egdes.index([u, v]))

我的解答思路和尝试过的方法

尝试过两个tensor都转为list,遍历64个边,对每一个边,用.index()找到它在1万个边的位置,这样做太慢了,有没有可以批量处理的方法?

  • 写回答

2条回答

      报告相同问题?

      相关推荐 更多相似问题

      问题事件

      • 系统已结题 10月14日
      • 已采纳回答 10月6日
      • 创建了问题 7月28日

      悬赏问题

      • ¥20 怎么将vs中的程序同步到sql中的表中
      • ¥15 用flask框架接收前端上传的文本文件后怎么读取该文件
      • ¥15 有没有嵌入式ubuntu的朋友
      • ¥50 diddler Burp Suite
      • ¥20 关于UE5.1的问题,求专家解答!
      • ¥30 关于#爬虫#的问题:百度为您找到相关结果约100,000,000个 但只展现70页 有没有可能提取出来全部的搜索结果
      • ¥40 IIS部署API,弄了几天没弄出来,希望带佬远程帮我弄一下,有偿
      • ¥15 关于#c##的问题:winform中利用delegate更新ui的顺序错误
      • ¥15 有没有985信息安全专业的同学说一下你们都学什么。。
      • ¥15 qt5.6.1读取oracle8中文数据乱码