В lua/torch запишите тензор без печатного типа

Я новичок в lua и просто не могу найти ответ на то, что кажется ужасно простым вопросом.

Я хочу распечатать некоторые тензоры, которые соответствуют вложениям слов в стиле Word2Vec. Каждая строка должна начинаться со слова, за которым следуют элементы тензора. У меня есть следующий код:

function Word2Vec:print_semantic_space()
    if self.word_vecs_norm == nil then
        self.word_vecs_norm = self:normalize(self.word_vecs.weight:double())
    end
    for word,_ in pairs(self.vocab) do
        vec=self.word_vecs_norm[self.word2index[word]]
        vec:resize(vec:size(1),1) 
        vec=vec:t()
        io.write(word," ",tostring(vec))
    end
end

Это все хорошо и хорошо, но я также продолжаю распечатывать тип и размер тензора:

usually -0.2063  0.5654  0.1447  0.2765 -0.3903  0.2646  0.2254  0.5064 -0.1009 -0.0260
[torch.DoubleTensor of size 1x10]
go -0.5896  0.1330  0.1361 -0.0193 -0.5612  0.3529  0.3683  0.0141  0.0447 -0.1963
[torch.DoubleTensor of size 1x10]

Как я могу сказать lua не возвращать тип? Как это:

usually -0.2063  0.5654  0.1447  0.2765 -0.3903  0.2646  0.2254  0.5064 -0.1009 -0.0260
go -0.5896  0.1330  0.1361 -0.0193 -0.5612  0.3529  0.3683  0.0141  0.0447 -0.1963

Извините, если ответ уже есть, и я не искал нужные ключевые слова. Я все еще новичок в концепциях lua.


torch lua
person linguist    schedule 28.03.2016    source источник

Ответы (1)


arrow_upward
0
arrow_downward

Вы можете написать свою собственную функцию дампа, например:

local dump = function(vec)
    vec = vec:view(vec:nElement())
    local t = {}
    for i=1,vec:nElement() do
        t[#t+1] = string.format('%.4f', vec[i])
    end
    return table.concat(t, '  ')
end

И используйте его вместо tostring.

person deltheil    schedule 29.03.2016