pytorch argsort

亦凉 2022-09-04 14:49 117阅读 0赞

**维度1上倒叙排序：**

方法1，先排序，再反转

index_score=score_mat.argsort(dim=1).flip(dims=[1])

方法2 倒序排序

index_score=score_mat.argsort(dim=1,descending=True)

tok排序：

pred = torch.randn((4, 4))
        print(pred)
        values, indices = pred.topk(3, dim=1)
        print(indices)

torch.topk(input, k, dim=None, largest=True, sorted=True, out=None) -> (Tensor, LongTensor)

参数:

input (Tensor) – 输入张量  
k (int) – “top-k”中的k  
dim (int, optional) – 排序的维  
largest (bool, optional) – 布尔值，控制返回最大或最小值  
sorted (bool, optional) – 布尔值，控制返回值是否排序  
out (tuple, optional) – 可选输出张量 (Tensor, LongTensor) output buffer

经过测试，效果一样的：

import torch
    
    data=torch.randn(4,4)
    
    print(data)
    
    index_score=data.argsort(dim=1).flip(dims=[1])
    
    print(index_score)
    
    
    index_score2=data.argsort(dim=1,descending=True)
    
    print(index_score2)

pytorch反转：

import torch
    
    data=torch.arange(10).view(2, 5)
    print(data)
    aaa=torch.flip(data, dims=[ 1])
    
    print(aaa)
    
    bbb=torch.flip(data, dims=[0])	# 对第0维进行反转
    
    print(bbb)
    ccc= torch.flip(data, dims=[0, 1])	# 对第0、1维进行反转
    
    print(ccc)
    
    ddd= data.flip(dims=[0, 1])	# 对第0、1维进行反转，与上一句效果相同
    
    print(ddd)

发表评论取消回复

表情：

评论列表（有 0 条评论，117人围观）

还没有评论，来说两句吧...

相关阅读

相关 np.argsort()函数

定义一个一维数组 >>> a=[2,-1,4] >>> a=np.array(a) >>> b=np.argsort(a) 返回的是数组a中从

阳光穿透心脏的1/2处/ 2023年10月04日 14:59/ 0 赞/ 7 阅读

相关 python numpy argsort 白话解释

大家在使用numpy的时候可能遇到这样的代码： argsort() 官方给的这个函数的解释是一句有些让人迷惑的话: > Returns the indices tha

缺乏、安全感/ 2022年10月08日 13:25/ 0 赞/ 116 阅读

相关 pytorch argsort

维度1上倒叙排序：方法1，先排序，再反转 index_score=score_mat.argsort(dim=1).flip(dims=[1]) 方法2 倒序排

亦凉/ 2022年09月04日 14:49/ 0 赞/ 118 阅读

相关 numpy中argsort函数用法

在Python中使用help帮助 >>> import numpy >>> help(numpy.argsort) Help on function argsort

落日映苍穹つ/ 2022年08月26日 01:26/ 0 赞/ 137 阅读

相关 sort() , sorted() 与argsort()

sort() 是可变对象的（字典、列表）方法，无参数、无返回值，因为它可以改变对象，因此无需返回值。sort()方法对于不可变对象（如元组、字符串）是不能调用的。 ![这里

本是古典何须时尚/ 2022年07月26日 05:28/ 0 赞/ 211 阅读

相关 python中argsort的使用-排序

在[Python][]中使用help帮助 >>> import numpy >>> help(numpy.argsort) Help on function args

﹏ヽ暗。殇╰゛Y/ 2022年07月15日 18:22/ 0 赞/ 316 阅读

相关 Numpy之argsort函数用法

文档原文： [https://docs.scipy.org/doc/numpy/reference/generated/numpy.argsort.html][https_

我就是我/ 2022年06月10日 04:29/ 0 赞/ 205 阅读

相关 Numpy-argsort函数

argsort是numpy模块中的类。y=argsort(x)将x中的元素从小到大排列，提取其对应的index，然后输出到y。例如： x=np.array([2,6

Bertha 。/ 2022年06月04日 03:28/ 0 赞/ 178 阅读

相关 C++11实现argsort功能

1. 编码 // 实现argsort功能 template<typename T> std::vector<int> argsort(const std:

浅浅的花香味﹌/ 2022年03月28日 15:20/ 0 赞/ 295 阅读

相关 numpy argsort

import numpy as np x = np.array([3,1,2]) np.argsort(x) > 输出 > array(\[1,2

Dear 丶/ 2021年11月15日 13:12/ 0 赞/ 293 阅读