你即將離開本站

並前往https://cloud.tencent.com/developer/article/1755705

PyTorch入门笔记-index_select选择函数- 云+社区- 腾讯云

接下来使用torch.index_select(input,dim,index,out=None) 函数分别对1D 张量、2D 张量和3D 张量进行索引。 >>> import torch >>> # 创建1D张量>>> a ...

確定！ 回上一頁

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