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

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条回答 默认 最新

  • 乄洛尘 人工智能领域优质创作者 2022-07-29 12:38
    关注

    法一:遍历64个边,每边 用 repeat 变成 all_edges 的形状,然后利用“==” ,取出为 True 的索引。
    法二:将 all_edges:tensor shape[2,10640] 增加一个维度变成 all_edges:tensor shape[2, 1, 10640] ,再 repeat,变成 all_edges:tensor shape[2, 64, 10640]
    将 target_nodes:tensor shape[2,64] 变成 target_nodes:tensor shape[2,64, 1],再 repeat,变成 target_nodes:tensor shape[2,64, 10640],
    两者 "==" 取出为 True 的索引,反着推就能推出要得到的2维索引了

    本回答被题主选为最佳回答 , 对您是否有帮助呢?
    评论 编辑记录
查看更多回答(1条)

报告相同问题?

问题事件

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

悬赏问题

  • ¥15 #MATLAB仿真#车辆换道路径规划
  • ¥15 java 操作 elasticsearch 8.1 实现 索引的重建
  • ¥15 数据可视化Python
  • ¥15 要给毕业设计添加扫码登录的功能!!有偿
  • ¥15 kafka 分区副本增加会导致消息丢失或者不可用吗?
  • ¥15 微信公众号自制会员卡没有收款渠道啊
  • ¥100 Jenkins自动化部署—悬赏100元
  • ¥15 关于#python#的问题:求帮写python代码
  • ¥20 MATLAB画图图形出现上下震荡的线条
  • ¥15 关于#windows#的问题:怎么用WIN 11系统的电脑 克隆WIN NT3.51-4.0系统的硬盘