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

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

无论如何，如果你想要将这些信息保存在一个字符串中，你可能需要自己提取它们。这很容易做，只需要使用一个简单的循环：

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

如果你想要漂亮的打印输出，我建议你看一下 TREPL 的代码（就像评论中 @nobody 建议的那样）。它们全部都是 Lua 代码，所以很容易复制。更确切地说，我推荐他们的 sizestr()、print_new() 和 printvar() 函数。只需将它们修改一下，使其构建字符串而不是打印输出即可。

下面是使用他们 sizestr() 函数的一个例子：

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

这输出和你所希望的一样，同时也构建了一个字符串。这远远不是最优的解决方案，但它是个好的开始。