pytorch函数之torch.index_select

torch.index_select函数，顾名思义就是根据index索引在input输入张量中选择某些特定的元素，下面介绍该函数的参数。python torch.index_select(input, dim, index, out=None):web input：输入Tensor，在该Tensor上根据index和dim进行切片； dim：切片维度，在input这个Tensor的哪一个维度上进行

>>阅读原文<<