torch.sort()

torch.sort()，对tensor中元素排序

用法

dim = -1，按照行排序，dim= 1按照列排序，descending=True,则递减排序，否则递增

按照行排序

logits = torch.tensor([[[-0.5816, -0.3873, -1.0215, -1.0145,  0.4053],[ 0.7265,  1.4164,  1.3443,  1.2035,  1.8823],[-0.4451,  0.1673,  1.2590, -2.0757,  1.7255],[ 0.2021,  0.3041,  0.1383,  0.3849, -1.6311]]])sorted_logits, sorted_indices = torch.sort(logits, descending=True, dim=-1)  # 对logits进行递减排序
print(sorted_logits)
print(sorted_indices)
输出：
tensor([[[ 0.4053, -0.3873, -0.5816, -1.0145, -1.0215],[ 1.8823,  1.4164,  1.3443,  1.2035,  0.7265],[ 1.7255,  1.2590,  0.1673, -0.4451, -2.0757],[ 0.3849,  0.3041,  0.2021,  0.1383, -1.6311]]])
tensor([[[4, 1, 0, 3, 2],[4, 1, 2, 3, 0],[4, 2, 1, 0, 3],[3, 1, 0, 2, 4]]])

按照列排序

sorted_logits, sorted_indices = torch.sort(logits, descending=True, dim=1)  # 对logits进行递减排序
print(sorted_logits)
print(sorted_indices)
输出：
tensor([[[ 0.7265,  1.4164,  1.3443,  1.2035,  1.8823],[ 0.2021,  0.3041,  1.2590,  0.3849,  1.7255],[-0.4451,  0.1673,  0.1383, -1.0145,  0.4053],[-0.5816, -0.3873, -1.0215, -2.0757, -1.6311]]])
tensor([[[1, 1, 1, 1, 1],[3, 3, 2, 3, 2],[2, 2, 3, 0, 0],[0, 0, 0, 2, 3]]])

torch.sort()相关推荐

PyTorch 笔记（08）— Tensor 比较运算（torch.gt、lt、ge、le、eq、ne、torch.topk、torch.sort、torch.max、torch.min）
1. 常用函数比较函数中有一些是逐元素比较,操作类似逐元素操作,还有一些类似归并操作,常用的比较函数如下表所示. 表中第一行的比较操作已经实现了运算符重载,因此可以使用 a>=b,a>b ...
代码阅读 | torch.sort函数
知识点https://blog.csdn.net/u012495579/article/details/106117511网友讲得非常清晰了. 代码如下: lengths = [17,17,19,23 ...
torch.sort()用法
torch.sort()用法 torch.sort(input, dim=- 1, descending=False, stable=False, out=None) 这个函数用于将输入的tensor ...
pytorch torch.sort()的用法以及例子
1.作用对tensor中元素排序 2.用法 dim = -1,按照行排序,dim= 1按照列排序,descending=True,则递减排序,否则递增 3.例子按照行排序 logits = tor ...
学习笔记CB012: LSTM 简单实现、完整实现、torch、小说训练word2vec lstm机器人
摘要: 真正掌握一种算法,最实际的方法,完全手写出来. LSTM(Long Short Tem Memory)特殊递归神经网络,神经元保存历史记忆,解决自然语言处理统计方法只能考虑最近n个词语而忽略更 ...
torch 深度学习(3)
torch 深度学习(3) 损失函数,模型训练前面我们已经完成对数据的预处理和模型的构建,那么接下来为了训练模型应该定义模型的损失函数,然后使用BP算法对模型参数进行调整损失函数 Criterio ...
Torch深度学习的60分钟教程（翻译）
这篇博客是官方Tutorial的翻译(加一丢丢自己的见解吧),原文链接为https://github.com/soumith/cvpr2015/blob/master/Deep%20Learning% ...
python torch exp_PyTorch官方中文文档：torch
torch 包 torch 包含了多维张量的数据结构以及基于其上的多种数学操作.另外,它也提供了多种工具,其中一些可以更有效地对张量和任意类型进行序列化. 它有CUDA 的对应实现,可以在NVIDIA ...
小白学Pytorch系列--Torch API (7)
小白学Pytorch系列–Torch API (7) Comparison Ops allclose 此函数检查输入和其他是否满足条件: >>> torch.allclose(tor ...
pytorch常用代码
20211228 https://mp.weixin.qq.com/s/4breleAhCh6_9tvMK3WDaw 常用代码段本文代码基于 PyTorch 1.x 版本,需要用到以下包: impo ...

torch.sort()

用法

torch.sort()相关推荐

最新文章

热门文章