你即將離開本站

並前往https://zhuanlan.zhihu.com/p/352877584

图解PyTorch中的torch.gather函数 - 知乎专栏

GATHER官方文档对torch.gather()的定义非常简洁定义：从原tensor中获取指定di… ... Compute Q(s_t, a) - the model computes Q(s_t), then we select the # columns of ...

確定！ 回上一頁

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

  1. Torch unsqueeze
    2.
  2. Torch gather
    3.
  3. Torch nonzero
    4.
  4. Torch permute
    5.
  5. Torch expand
    6.
  6. Python gather
    7.
  7. Pytorch dimension
    8.
  8. Torch tensor
    9.