概述
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) ，即从小到大排序
最后
以上就是朴实小鸭子为你收集整理的pytorch中sort的简单用法sort()在pytorch中使用的全部内容，希望文章能够帮你解决pytorch中sort的简单用法sort()在pytorch中使用所遇到的程序开发问题。
如果觉得靠谱客网站的内容还不错，欢迎将靠谱客网站推荐给程序员好友。
本图文内容来源于网友提供，作为学习参考使用，或来自网络收集整理，版权属于原作者所有。
发表评论 取消回复