py argsort()函数的用法

python argsort()函数

该函数作用到一个数组,也可以是np数据或torch数据

作用后得到一个同维度的数据,每个位置的元素为原数组其元素在所在排的大小次序。

为了更好的解释这个函数,我们用到torch.randperm(n)函数,可以生成一个由0到n-1组成的n维tensor数据

在这里插入图片描述
比如我们得到的x其最大值在第1个位置(从0计数),那么1就在x.argsort()的最后一个位置。
x的最小值在第三个位置,那么3就在x.argsort()的第0个位置。

输入的是矩阵,其输出同理:
在这里插入图片描述
其中stack函数可以按照指定维度堆叠把tensor堆在一起。

点赞是一件有意义的事情,不仅是对作者,也对读者。
如果时光可以重来,我一定会给那些指引过我的文章点赞

猜你喜欢

转载自blog.csdn.net/fei_YuHuo/article/details/114736411
py