从torch.Tensor中删除项目
2017-7-29 12:7:35
收藏:0
阅读:87
评论:1
我有以下用Lua编写的代码。
我想从scores中获取N个最高分数的索引及其相应的分数。
似乎我需要逐步从scores中删除当前最大值并重新获取最大值,但找不到适当的方法来完成。
nqs=dataset['question']:size(1);
scores=torch.Tensor(nqs,noutput);
qids=torch.LongTensor(nqs);
for i=1,nqs,batch_size do
xlua.progress(i, nqs)
r=math.min(i+batch_size-1,nqs);
scores[{{i,r},{}}],qids[{{i,r}}]=forward(i,r);
-- print(scores)
end
tmp,pred=torch.max(scores,2);
点赞
评论区的留言会收到邮件通知哦~
推荐文章
- Lua 虚拟机加密load(string.dump(function)) 后执行失败问题如何解决
- 我想创建一个 Nginx 规则,禁止访问
- 如何将两个不同的lua文件合成一个 东西有点长 大佬请耐心看完 我是小白研究几天了都没搞定
- 如何在roblox studio中1:1导入真实世界的地形?
- 求解,lua_resume的第二次调用继续执行协程问题。
- 【上海普陀区】内向猫网络招募【Skynet游戏框架Lua后端程序员】
- SF爱好求教:如何用lua实现游戏内调用数据库函数实现账号密码注册?
- Lua实现网站后台开发
- LUA错误显式返回,社区常见的规约是怎么样的
- lua5.3下载库失败
- 请问如何实现文本框内容和某个网页搜索框内容连接,并把网页输出来的结果反馈到另外一个文本框上
- lua lanes多线程使用
- 一个kv数据库
- openresty 有没有比较轻量的 docker 镜像
- 想问一下,有大佬用过luacurl吗
- 在Lua执行过程中使用Load函数出现问题
- 为什么 neovim 里没有显示一些特殊字符?
- Lua比较两个表的值(不考虑键的顺序)
- 有个lua简单的项目,外包,有意者加微信 liuheng600456详谈,最好在成都
- 如何在 Visual Studio 2022 中运行 Lua 代码?

希望我没有误解,因为你展示的代码(特别是循环)似乎并不与你想要做的事情相关。无论如何,以下是我会这样做。
sr=scores:view(-1,scores:size(1)*scores:size(2)) val,id=sr:sort() --val是一个按升序存储值的行向量 --id将是sr中对应的索引 --现在您可以从末尾切片val和id以找到您想要的N个值,然后使用以下方法仅仅是恢复在分数矩阵中的原始指数: col=(index-1)%scores:size(2)+1 row=math.ceil(index/scores:size(2))希望这会有帮助。