lua 火炬，如何将 print() 输出转换为字符串答案

【问题标题】：lua torch, how to get the print() output into a stringlua 火炬，如何将 print() 输出转换为字符串
【发布时间】：2017-06-25 19:58:56
【问题描述】：

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
}

我想获取此打印输出以用于记录目的。有谁知道这是怎么做到的吗？

【问题讨论】：

我不知道torch 但试试tostring(criterion)。
tostring(criterion) 只提供nn.ClassNLLCriterion，遗憾的是没有额外的信息
如果没有人知道，那么也许看看他们如何重新定义 print 或者是否可以哄骗标准的 __tostring（如果事情仍然这样）吐出长版。这可能需要对源头进行一些挖掘，但应该是可行的。尝试查看debug.getinfo( getmetatable( criterion ).__tostring, "S" ) 是否有效并产生一些有用的信息（你会得到一个包含信息的表格，如果__tostring 是用 Lua 编写的，那会给你文件和行范围；如果是 C，它只会说“[C]”）。（也许在 print 上尝试相同的 getinfo。）

标签： lua neural-network torch

【解决方案1】：

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

无论如何，如果您想在字符串中包含此信息，您可能必须自己提取它。这可以通过一个简单的循环轻松完成：

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

如果您想要精美的打印输出，那么我建议您查看 TREPL 的代码（就像 cmets 中建议的 @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)

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

【讨论】：