indices should be either on cpu or on the same device as the indexed tensor (cpu) indices should be either on cpu or on the same device as the indexed tensor (cpu) 深度学习 2022-11-23 2 点赞 0 评论 7058 阅读