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)

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

lua torch，如何将print（）输出到字符串中

1 个答案: