c++ - Libtorch C++ - 没有匹配的成员函数用于调用 InterpolateFuncOptions 的 'size'

Question

在 C++ 中使用 Libtorch 1.6.0，我收到以下错误:

error: no matching member function for call to 'size'

我的线路如下:

image = F::interpolate(image, F::InterpolateFuncOptions().size({target_height, target_width}).mode(torch::kNearest));

但在文档中它似乎是正确的......知道吗？
提前致谢

Answer 1

你应该用 std::vector 包裹它像这样:

image = F::interpolate(image, 
        F::InterpolateFuncOptions()
        .size(std::vector<>{target_height, target_width})
        .mode(torch::kNearest));

原因是 size没有对 std::initializer_list 的重载调用您尝试使用的(参见 size 文档 here)。