如何从 libtorch 输出中移除乘数并显示最终结果？

Question

当我尝试 display/print 将一些张量显示到屏幕上时，我遇到了类似下面的内容，而不是得到最终结果，libtorch 似乎显示了带有乘数的张量（即 0.01*以及您在下面看到的类似内容）：

offsets.shape: [1, 4, 46, 85]
probs.shape: [46, 85]
offsets: (1,1,.,.) =
 0.01 *
  0.1006  1.2322
  -2.9587 -2.2280

(1,2,.,.) =
 0.01 *
  1.3772  1.3971
  -1.2813 -0.8563

(1,3,.,.) =
 0.01 *
  6.2367  9.2561
   3.5719  5.4744

(1,4,.,.) =
  0.2901  0.2963
  0.2618  0.2771
[ CPUFloatType{1,4,2,2} ]
probs: 0.0001 *
 1.4593  1.0351
  6.6782  4.9104
[ CPUFloatType{2,2} ]

如何禁用此行为并获得最终输出？我试图将其显式转换为浮点数，希望这会导致最终输出为 stored/displayed 但这也不起作用。

Answer 1

根据libtorch输出张量的源代码，在版本库中搜索“*”字符串后，发现这个“pretty-print”是在aten/src/ATen/core/Formatting.cpp翻译单元中完成的.刻度和星号在此处添加：

static void printScale(std::ostream & stream, double scale) {
  FormatGuard guard(stream);
  stream << defaultfloat << scale << " *" << std::endl;
}

然后张量的所有坐标都除以scale:

if(scale != 1) {
  printScale(stream, scale);
}
double* tensor_p = tensor.data_ptr<double>();
for(int64_t i = 0; i < tensor.size(0); i++) {
  stream << std::setw(sz) << tensor_p[i]/scale << std::endl;
}

基于这个翻译单元，这根本不可配置。

我猜你有两个选择：

调整功能并尽可能少地编辑现有功能以满足您的要求。
删除（或添加 #ifdef）Formatting.cpp 中 Tensor 的 << 运算符重载并提供您自己的实现。但是，在构建 libtorch 时，您必须 link 将其指向包含该方法实现的目标。

但是，这两种选择都需要您更改第 3 方代码，我认为这很糟糕。

如何从 libtorch 输出中移除乘数并显示最终结果？

How to remove the multiplier from the libtorch output and display the final result?

c++

torch

libtorch