pytorch函數之torch.index_select

torch.index_select函數，顧名思義就是根據index索引在input輸入張量中選擇某些特定的元素，下面介紹該函數的參數。python torch.index_select(input, dim, index, out=None):web input：輸入Tensor，在該Tensor上根據index和dim進行切片； dim：切片維度，在input這個Tensor的哪一個維度上進行

>>阅读原文<<