你即將離開本站

並前往https://www.3water.com/article/4MTAx9NzM40LmQ1

pytorch中index_select()的用法详解- 3WATER - 三水点靠木

a = torch.linspace(1, 12, steps=12).view(3, 4); print(a); b = torch.index_select(a, 0, torch.tensor([0, 2])); print(b) ...

確定！ 回上一頁

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