lua-torch，如何将print()输出转换为字符串

Question

require 'nn'
criterion = nn.ClassNLLCriterion()
print(criterion)

该输出

nn.ClassNLLCriterion
{
sizeAverage : true
output : 0
gradInput : DoubleTensor - empty
output_tensor : DoubleTensor - size: 1
target : LongTensor - size: 1
total_weight_tensor : DoubleTensor - size: 1
}

我想得到这个打印输出用于日志记录。有人知道怎么做吗？

Answer 1

嗯，我不确定你是如何得到这个输出的。当我运行这个代码时，我得到了nn.ClassNLLCriterion。也许是不同的Lua/Torch版本？

无论如何，如果你想把这些信息放在一个字符串中，你可能必须自己提取。这可以很容易地完成一个简单的循环：

for k,v in pairs(criterion) do
print(k,v)
end

如果你想要精美的打印输出，那么我建议你看看TREPL的代码(比如评论中的@nobody)。这一切都在Lua，所以很容易复制。更确切地说，我推荐它们的sizestr()、print_new()和printvar()函数。只需更改它们，它们就可以构建一个字符串，而不是打印。

使用sizestr函数的一个快速示例：

-- Copy/Paste from trepl/init.lua
local function sizestr(x)
local strt = {}
if _G.torch.typename(x):find('torch.*Storage') then
return _G.torch.typename(x):match('torch%.(.+)') .. ' - size: ' .. x:size()
end
if x:nDimension() == 0 then
table.insert(strt, _G.torch.typename(x):match('torch%.(.+)') .. ' - empty')
else
table.insert(strt, _G.torch.typename(x):match('torch%.(.+)') .. ' - size: ')
for i=1,x:nDimension() do
table.insert(strt, x:size(i))
if i ~= x:nDimension() then
table.insert(strt, 'x')
end
end
end
return table.concat(strt)
end

local function sutoringu(elem)
local str = ''
if torch.isTensor(elem) then
str = sizestr(elem)
else
str = tostring(elem)
end
return str
end
local str = '{n'
local tab = '  '
for k,v in pairs(criterion) do
str = str .. tab .. k .. ' : ' .. sutoringu(v) .. 'n'
end
str = str .. '}'
print(str)

这将输出与您希望的内容相同的内容，即在过程中构造一个字符串。这远不是最佳的，但这只是一个开始。