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 评论 8531 阅读