[人工智能]pytorch argsort

维度1上排序，并且倒叙输出。

index_score=score_mat.argsort(axis=1).flip(dims=[1])

index_score=score_mat.argsort(axis=1,descending=True)

经过测试，效果一样的：

import torch

data=torch.randn(4,4)

print(data)

index_score=data.argsort(axis=1).flip(dims=[1])

print(index_score)


index_score2=data.argsort(axis=1,descending=True)

print(index_score2)

pytorch反转，不是倒叙：

import torch

data=torch.arange(10).view(2, 5)
print(data)
aaa=torch.flip(data, dims=[ 1])

print(aaa)

bbb=torch.flip(data, dims=[0])	# 对第0维进行反转

print(bbb)
ccc= torch.flip(data, dims=[0, 1])	# 对第0、1维进行反转

print(ccc)

ddd= data.flip(dims=[0, 1])	# 对第0、1维进行反转，与上一句效果相同

print(ddd)