可以看出gather的作用，index是索引，具体是行还是列的索引要看dim 的指定，指定dim=1，也就是横向，那么索引就是列号。指定dim=0，也就是纵向，那么索引就是行号。
pytorch中gather函数解析
本文详细解析了PyTorch中的gather函数作用，通过示例说明了如何使用该函数根据索引从张量中收集元素。gather函数的索引依据dim参数决定，dim=1时按列索引，dim=0时按行索引。
扫一扫
评论
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
查看更多评论
添加红包
