pytorch中sort的简单用法

sort()在pytorch中使用在numpy中有 sort 和 sorted 两种，但不能对索引值排序。# pytorch中使用import torch# 已知一个tensor，如，dj 是 tenor([1,3,4,2])a , b = dj.sort(0,True)# True是从大到小排序# 得到的a 为 tensor([1,2,3,4])# b 为 tensor([0,3,1,2])#

murphys_law_dj

1043人浏览 · 2021-04-28 20:30:36

murphys_law_dj · 2021-04-28 20:30:36 发布

sort()在pytorch中使用

在numpy中有 sort 和 sorted 两种，但不能对索引值排序。

# pytorch中使用
import torch
# 已知一个tensor，如，dj 是 tenor([1,3,4,2])
a , b = dj.sort(0,True)  # True是从大到小排序
# 得到的a 为 tensor([1,2,3,4])
# b 为 tensor([0,3,1,2])
# 当然，sort()括号内为空等价于 (0,False) ，即从小到大排序

https://edu.csdn.net/learn/39067/627173?utm_source=2019755004

汇聚全球AI编程工具，助力开发者即刻编程。

更多推荐

DeepSeek官方推荐的DeepCode，值不值得装？一次性讲清楚！

AI编程社区

【从零学Vibe Coding】第六章：主流 Vibe Coding 工具大全

本文系统梳理了2025年主流AI编程工具的特点与适用场景。重点介绍了6款工具：Cursor（深度整合AI的IDE）、GitHub Copilot（GitHub生态插件）、Claude Code（终端AI工程师）、ChatGPT（综合思考伙伴）、Windsurf（强调代理协作的IDE）和Trae（激进的多Agent IDE）。通过对比表格和决策树，帮助开发者根据自身需求选择工具：新手建议从ChatG