移动端[Android]上的完整c++ LibTorch API

Question

我正在尝试编译一个使用LibTorch for android的程序。但是在pytorch的repo中，script/build_android.sh构建的库只包含torch.jit模块。我需要torch.nn模块

是否可以访问完整的c++ LibTorch API ?如果有，我该如何构建它?

Answer 1

似乎通过在根CMake文件中设置NO_API为OFF和BUILD_MOBILE_AUTOGRAD为ON，您可以构建api，从而构建torch.nn。