摘要: 原贴地址:https://www.jianshu.com/p/e568213c8501 写模型的时候需要根据索引赋,直接写python效率很低,看到这个帖子原来pytorch内部已经实现了。 主要是 scatter index_fill index_put 阅读全文
posted @ 2021-06-17 13:14 Luke_Ye 阅读(655) 评论(0) 推荐(0) 编辑