首页 新闻 会员 周边 捐助
0
回答数

待解决问题 pytroch,返回tensor中多个特定值的索引

想从所有的边(1万多个)中找到指定64个边的索引 所有的边all_edges: tensor shape[2,10640] 指定边:target_nodes: tensor shape [2,64 ]