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

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

想从所有的边（1万多个）中找到指定64个边的索引 所有的边all_edges: tensor shape[2,10640] 指定边：target_nodes: tensor shape [2,64 ]
pytorch 深度学习 机器学习 python
为什么要学习的主页 为什么要学习
浏览(144) 2年前