pytorch index_select用法

        在看動手學深度學習 pytorch版本的時候,看到其中使用了index_select方法。感覺這個方法較爲常用和有用。所以需要弄懂。 用法如下:         torch.index_select(input, dim, index, out=None) → Tensor 參數input(Tensor):表示被選擇的tensor 參數dim(int):表示是在哪個維度做選擇。爲1
相關文章
相關標籤/搜索