你即將離開本站

並前往https://www.796t.com/article.php?id=9917

Pytorch中index_select() 函式的實現理解 - 程式人生

... index：表示從第一個引數維度中的哪個位置挑選資料，型別為torch.Tensor類的例項；. 剛開始學習pytorch，遇到了index_select(),一開始不太明白幾個引數的意思， ...

確定！ 回上一頁

查詢 「Torch index_select」的人也找了：

  1. Torch gather
    2.
  2. Torch where index
    3.
  3. Torch nonzero
    4.
  4. Torch gather 3d
    5.
  5. Torch take
    6.
  6. Torch permute
    7.
  7. Torch masked_select
    8.
  8. Torch-scatter
    9.