分类：gather方法

2021
07-16

Pytorch中的gather使用方法

官方说明gather可以对一个Tensor进行聚合，声明为：torch.gather(input,dim,index,out=None)→Tensor一般来说有三个参数：输入的变量input、指定在某一维上聚合的dim、聚合的使用的索引index，输出为Tensor类型的结果（index必须为LongTensor类型）。#参数介绍：input(Tensor)?Thesourcetensordim(int)?Theaxisalongwhichtoindexindex(LongTensor)?Theindicesofelementstogatherout(Tensor,optional)... 继续阅读 >