你即將離開本站

並前往https://runebook.dev/zh-CN/docs/pytorch/generated/torch.gather

torch.gather - 沿dim 所指定的轴收集值。 对于3-D张量来说 ...

torch.gather(input, dim, index, *, sparse_grad=False, out=None) → Tensor. 沿 dim 所指定的轴收集值。 对于3-D张量来说,输出由以下方式指定。

確定！ 回上一頁

查詢 「Torch gather」的人也找了：

  1. Torch-scatter
    2.
  2. Torch gather
    3.
  3. Torch gather 用法
    4.
  4. Torch gather dimension
    5.
  5. Python gather
    6.
  6. Torch gather 3d
    7.
  7. Torch where index
    8.
  8. Torch .max outputs.data, 1
    9.