torch.argsort¶
- torch.argsort(input, dim=-1, descending=False, stable=False) Tensor ¶
返回按值升序对给定维度上的张量进行排序的索引。
这是
torch.sort()
返回的第二个值。有关此方法的精确语义,请参阅其文档。如果
stable
为True
,则排序例程变为稳定,保持等效元素的顺序。如果为False
,则比较相等的值的相对顺序不受保证。True
速度较慢。- 参数
示例
>>> a = torch.randn(4, 4) >>> a tensor([[ 0.0785, 1.5267, -0.8521, 0.4065], [ 0.1598, 0.0788, -0.0745, -1.2700], [ 1.2208, 1.0722, -0.7064, 1.2564], [ 0.0669, -0.2318, -0.8229, -0.9280]]) >>> torch.argsort(a, dim=1) tensor([[2, 0, 3, 1], [3, 2, 1, 0], [2, 1, 0, 3], [3, 2, 1, 0]])