2025年12月3日
Pytorch随笔：gather函数
摘要： gather函数使用方法： gather(dim,index) dim = 0时，使用index填充行；dim = 1时，使用index填充列。 例如，index = tensor([[x,y,z]]) gather(dim=0,index)示意图为： tensor([[(x,0),(y,1),(z 阅读全文
posted @ 2025-12-03 15:32 关注永雏塔菲谢谢喵 阅读(2) 评论(0) 推荐(0)