pytorch index_select()函數

函數實現從當前張量中從某個維度選擇一部分序號的張量

tensor.select_index(dim, index)
對於一個二維張量feature：

第一個參數

參數0表示按行索引，1表示按列進行索引

第二個參數

是一個整數類型的一維tensor，就是索引的序號

二維張量舉例：

三維張量舉例：

另一種使用方式：

torch.select_index(tensor, dim, index)