pytorch函数 torch.index_select( )学习

y = torch.index_select(x, dim=1, index=index)

该函数是按照维度对原张量进行切分
index 是切分的索引，通常是一个数字型列表
x是待切分的张量

假设 x = torch.rand(3,2,1) 且 index = torch.LongTensor([0])

tensor([[[0.3048],
         [0.0140]],
        [[0.6699],
         [0.3395]],
        [[0.1088],
         [0.9452]]])

tensor([0])   注意这里的index 必须为longTensor类型

y = torch.index_select(x, dim=1, index=index)

将张量x 按照第二个维度（dim从0开始算）进行切分，搜索到第一个元素（index=0）
得到

tensor([[[0.3048]],
        [[0.6699]],
        [[0.1088]]])