网站建设资讯

NEWS

网站建设资讯

关于踩坑argmax和argsrt-创新互联

最近在做一些项目的时候遇到一些问题,在此记录一下

成都创新互联公司是一家专注于成都网站制作、网站建设与策划设计,方山网站建设哪家好?成都创新互联公司做网站,专注于网站建设十载,网设计领域的专业建站公司;建站业务涵盖:方山等地区。方山做网站价格咨询:135182197921、argmax

numpy作为python的开源库,一直受到广大用户的青睐,numpy库中涉及到的argmax函数,其作用是返回数组中大值的索引。

当一个数组中有多个相同的大值的时候,numpy会返回第一个出现的大值索引;当使用pytorch的argmax时候,返回的是最后一个出现的大值索引;

import numpy as np

a = [1, 2, 3, 3]
b = np.array(a)
c = from_numpy(b)
print(argmax(b))  #输出为2
print(torch.argmax(c))  #输出为3

当然在libtorch中和pytorch的情况是一样的,在c++中使用vector的函数寻找大值索引返回的也是第一个出现的大值索引,和numpy相同。

2、argsort

在pytorch、libtorch和numpy中都存在argsort函数,但是对同一个数组进行排序并且此数组中存在许多相同的值的时候,三个库的排序是不一样的,他们之间的排序逻辑存在差异。

你是否还在寻找稳定的海外服务器提供商?创新互联www.cdcxhl.cn海外机房具备T级流量清洗系统配攻击溯源,准确流量调度确保服务器高可用性,企业级服务器适合批量采购,新人活动首月15元起,快前往官网查看详情吧


标题名称:关于踩坑argmax和argsrt-创新互联
网页URL:http://cdweb.net/article/jpejh.html