你即將離開本站

並前往https://daimajiaoliu.com/daima/6cb8b703f387c09

torch.gather()和torch.index_select说明 - 代码交流

FloatTensor of size 3x4] 8 9>>> indices = torch.LongTensor([0, 2]) 10>>> torch.index_select(x, 0, indices) 11 12 1.2045 2.4084 0.4001 1.1372 13 1.3635 ...

確定！ 回上一頁

查詢 「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.