【Pytorch】torch.sort()方法使用

最新推荐文章于 2024-08-15 21:20:47 发布

智绘山河

最新推荐文章于 2024-08-15 21:20:47 发布

阅读量1.8k

点赞数 1

分类专栏： # pytorch 文章标签： pytorch

本文链接：https://blog.csdn.net/meiqi0538/article/details/124370563

版权

pytorch 专栏收录该内容

14 篇文章 2 订阅

订阅专栏

官方文档地址：https://pytorch.org/docs/stable/generated/torch.sort.html?highlight=sort#torch.sort

形式：torch.sort(input, dim=- 1, descending=False, stable=False, *, out=None)

作用：沿给定维度按值升序对输入张量的元素进行排序。如果没有给出 dim，则选择输入的最后一个维度。如果descending 为True，则元素按值降序排序。如果 stable 为 True，则排序例程变得稳定，保持等效元素的顺序。返回 (values, indices) 的命名元组，其中值是排序后的值，索引是原始输入张量中元素的索引。

参数：

input (Tensor) – the input tensor.
dim (int, optional) – the dimension to sort along
descending (bool, optional) – controls the sorting order (ascending or descending)
stable (bool, optional) – makes the sorting routine stable, which guarantees that the order of equivalent elements is preserved.

返回值：
out (tuple, optional) – the output tuple of (Tensor, LongTensor) that can be optionally given to be used as output buffers

使用案例如下：

x = torch.randn(3, 4)
sorted, indices = torch.sort(x)
sorted, indices = torch.sort(x, 0)
x = torch.tensor([0, 1] * 9)
x.sort()
x.sort(stable=True)
# torch.return_types.sort(
#    values=tensor([0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1]),
#    indices=tensor([ 0,  2,  4,  6,  8, 10, 12, 14, 16,  1,  3,  5,  7,  9, 11, 13, 15, 17]))