想从所有的边（1万多个）中找到指定64个边的索引 所有的边all_edges: tensor shape[2,10640] 指定边：target_nodes: tensor shape [2,64 ]
尝试过：把两个tensor都转为list，遍历指定边，对每一个边，用list的.index()方法找到它在所有边中的索引，这样的话，每找一个边就要到1万多个边里去找，运行起来实在太慢了，请问有没有更加简单的方法？
