pytorch index_select()函数

函数实现从当前张量中从某个维度选择一部分序号的张量

tensor.select_index(dim, index)
对于一个二维张量feature：

第一个参数

参数0表示按行索引，1表示按列进行索引

第二个参数

是一个整数类型的一维tensor，就是索引的序号

二维张量举例：

三维张量举例：

另一种使用方式：

torch.select_index(tensor, dim, index)