代码之家 › 专栏 › 技术社区 › ytrewq

从火炬中删除项目。张量

torch lua

ytrewq · 技术社区 · 8 年前

我在lua中编写了以下代码。

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);

1 回复 | 直到 8 年前

Ash 8 年前

 sr=scores:view(-1,scores:size(1)*scores:size(2))
 val,id=sr:sort()
 --val is a row vector with the values stored in increasing order
 --id will be the corresponding index in sr
 --now you can slice val and id from the end to find the N values you want, then you can recover the original index in the scores matrix simply with
 col=(index-1)%scores:size(2)+1
 row=math.ceil(index/scores:size(2))

推荐文章

Daniel Wagner · 如何使用带有管道的luv?

10 月前

Naruto_333 · Roblox:函数不循环或添加“货币”值

10 月前

user28944982 · IUPLUA使用table.unpack将元素放入.gridbox不起作用

10 月前

shingo.nakanishi · 如何在Lua中使用方法扩展现有类型,类似于Swift扩展或JavaScript原型?

10 月前

jiandingzhe · 如何使用C闭包预编译和转储Lua二进制文件?

10 月前

Lukas Raby · 玩Tween后,零件位置被锁定

1 年前

PatoFlamejanteTV · 无法克隆和移动已克隆的模型。(Roblox工作室)

1 年前

Lukas Raby · Clone()在将其父级设置为StarterPack时似乎不起作用

1 年前

Noway · 为什么roblox工作室在桌子上走错了路

1 年前

localhorst · 索引处的铁路超高访问表条目

1 年前