关于踩坑argmax和argsrt-创新互联
最近在做一些项目的时候遇到一些问题,在此记录一下
成都创新互联公司是一家专注于成都网站制作、网站建设与策划设计,方山网站建设哪家好?成都创新互联公司做网站,专注于网站建设十载,网设计领域的专业建站公司;建站业务涵盖:方山等地区。方山做网站价格咨询:135182197921、argmaxnumpy作为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-创新互联
文章起源:http://abwzjs.com/article/jpejh.html