import torch >>> a = torch.rand(3,3) >>> a tensor([[0.8395, 0.2678, 0.5021], ... torch.sortの2つめの戻り値と同じ>>> idx = torch.argsort(a) ...
確定! 回上一頁