你即將離開本站

並前往https://qiita.com/dokkozo/items/0aec98192b120a2ae65f

pytorchのtensorをsort, 同じ順で別のtensorを並べ替え ... - Qiita

指定した次元の方向にsortする# sortする次元はdimで指定>>> sorted, idx = torch.sort(a, dim = -1) # デフォルトはdim = -1 (-1は最後の次元.

確定！ 回上一頁

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

  1. Torch sort
    2.
  2. Numpy sort
    3.
  3. Torch argsort
    4.
  4. Pytorch sort lambda
    5.
  5. Torch permute
    6.
  6. Torch topk
    7.
  7. Torch expand
    8.
  8. Tensor index
    9.