你即將離開本站

並前往https://stackoverflow.com/questions/53986301/how-can-i-select-single-indices-over-a-dimension-in-pytorch

How can I select single indices over a dimension in pytorch?

sequences[torch.arange(sequences.size(0)), indices].

確定！ 回上一頁

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