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堆在一起。

20210313

标签：torch,函数,py,用法,位置,维度,argsort,tensor
来源： https://blog.csdn.net/fei_YuHuo/article/details/114736411