你即將離開本站

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

PyTorch入门笔记-gather选择函数 - 腾讯云

input (Tensor) - 输入张量; dim (int) - 需要进行索引的轴; index (LongTensor) - 要采集元素的索引; sparse_grad (bool, optional) - 如果为True， ...

確定！ 回上一頁

查詢 「Pytorch tensor index」的人也找了：

  1. Python Tensor index
    2.
  2. Pytorch list index
    3.
  3. PyTorch tensor CUDA
    4.
  4. Torch nonzero
    5.
  5. Torch gather
    6.
  6. Python gather
    7.
  7. Pytorchscatter
    8.
  8. PyTorch tensor to GPU
    9.