[torch/lua] Move util.Set in to test.lua.

This commit is contained in:
huxuan 2016-05-30 12:57:38 +08:00
Родитель 90cca4b79f
Коммит 41f905bf35
2 изменённых файлов: 7 добавлений и 8 удалений

Просмотреть файл

@ -3,11 +3,16 @@
require 'torch'
mv = require('multiverso')
util = require('util')
local mv_test = torch.TestSuite()
local mv_tester = torch.Tester()
function Set(list)
local set = {}
for _, l in ipairs(list) do set[l] = true end
return set
end
function mv_test.testArray()
size = 100000
tbh = mv.ArrayTableHandler:new(size)
@ -31,7 +36,7 @@ function mv_test.testMatrix()
for i = 1, 20 do
row_ids = {0, 1, 5, 10}
row_ids_set = util.Set(row_ids)
row_ids_set = Set(row_ids)
tbh:add(torch.range(1, size))
data = torch.range(
row_ids[1] * num_col + 1,

Просмотреть файл

@ -31,10 +31,4 @@ function util.cdata2tensor(cdata, sizes, data_type)
return data
end
function util.Set(list)
local set = {}
for _, l in ipairs(list) do set[l] = true end
return set
end
return util