pytorch索引查找 index_select的例子

import torch  x = torch.linspace(1, 12, steps=12).view(3,4) print(x)indices = torch.LongTensor([0, 2])y = torch.index_select(x, 0, indices)print(y) z = torch.index_select(x, 1, indices)print(z) z = torch.index_select(y, 1, indices)print(z)

tensor([[ 1.,  2.,  3.,  4.],    [ 5.,  6.,  7.,  8.],    [ 9., 10., 11., 12.]])tensor([[ 1.,  2.,  3.,  4.],    [ 9., 10., 11., 12.]])tensor([[ 1.,  3.],    [ 5.,  7.],    [ 9., 11.]])tensor([[ 1.,  3.],    [ 9., 11.]])