官方说明gather可以对一个Tensor进行聚合,声明为:torch.gather(input, dim, index, out=None) → Tensor一般来说有三个参数:输入的变量input、指定在某一维上聚合 ...
確定! 回上一頁