你即將離開本站

並前往https://www.codeleading.com/article/53975090221/

【torch.index_select】利用下标提取tensor中的值 - 代码先锋网

torch.index_select(self: Tensor, dim: Union[str, None], index: Tensor) ：第一个Tensor是被操作的tensor，dim表示要操作的维度，index是在这个维度下提取的索引张 ...

確定！ 回上一頁

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