摘要: 根据索引查找值 torch.gather() https://pytorch.org/docs/stable/generated/torch.gather.html#torch.gather 根据索引修改函数的值 torch.scatter() https://pytorch.org/docs/st 阅读全文
posted @ 2022-04-07 15:35 风影旋新月 阅读(203) 评论(0) 推荐(0) 编辑