Pytorch中 torch.index_select(input, dim, index, out=None) → Tensor 用法
1.返回值和参数解释
按照官网的解释
return(Tensor) 该方法返回一个Tensor,输出结果是按照输入的Tensor进行索引得到
input(Tensor)–输入张量。(原Tensor,得到的结果就是从该Tensor中产生)
dim(int)–我们索引的维度。(常用值:0/1 表示:按行/列索引)
index(LongTensor)–包含要索引的索引的一维张量。(选择原Tensor某些行或者某些列)
out(Tensor,可选)–输出张量。
2...
原创
2020-07-28 22:14:11 ·
843 阅读 ·
0 评论