你即將離開本站

並前往https://www.py4u.net/discuss/175939

Index pytorch 4d tensor by values in 2d tensor - py4u

eI = I[..., None, None].expand(-1, -1, 1, X.size(3)) # make eI the same for the last dimension Y = torch.gather(X, dim=2, index=eI).squeeze().

確定！ 回上一頁

查詢 「Torch expand dim」的人也找了：

  1. Torch unsqueeze
    2.
  2. Torch expand
    3.
  3. Torch expand dim
    4.
  4. Torch squeeze
    5.
  5. Torchcat
    6.
  6. Torch size
    7.
  7. Pytorch dimension
    8.
  8. Torch reshape
    9.