pytorch argsort

AI視覺網奇 2021-08-15 03:55:28 阅读数:277

本文一共[544]字,预计阅读时长:1分钟~
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)

版权声明:本文为[AI視覺網奇]所创,转载请带上原文链接,感谢。 https://gsmany.com/2021/08/20210815035505465q.html