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

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

第一个参数

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

第二个参数

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

二维张量举例:

pytorch index_select()函数

三维张量举例:

pytorch index_select()函数
pytorch index_select()函数
pytorch index_select()函数

另一种使用方式:

torch.select_index(tensor, dim, index)
pytorch index_select()函数