import torch
x= torch.randn(4)
print("x:",x)
y, index1 = x.sort( descending=True)
print("y:",y)
print("index1:",index1)
z, index2= index1.sort()
print("z:",z)
print("index2:",index2)
运行结果:
index1:每个值代表排序后的tensor y的每个元素,在排序前的tensor x的索引。
如:1代表y 中0.7613在x中的索引为1
index2:排序前的tensor x 的每个值的大小序号。
如:3表示-0.9252在排序后为第3小(从0
开始计数,即最小),0代表0.7613是x中的最大值(第0),1代表0.5979排序后为第1。