你即將離開本站

並前往http://www.4k8k.xyz/article/weixin_44021237/97413200

Pytorch中torch.gather函数的理解_dwzhu-程序员资料

torch.gather(input, dim, index, out=None) → Tensor. input (Tensor) – 源张量 dim (int) – 索引的轴 index (LongTensor) – 聚合元素的下标

確定！ 回上一頁

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