Slide 14
Slide 14 text
Copyright © OPTiM Corp. All Right Reserved. 14
PyTorchのC++ API
コンセプト
• PyTorchのプロダクション用に見据えて用意されたもの
• LibTorch / TorchScriptはPyTorch -> ONNX -> Caffe2の置き換えを意図したもの
• セマンティクスの保持、開発者が二つの分断されたフレームワークを扱わなくてよい
基本機能
• PyTorch同様、学習 / 推論、CPU実行 / GPU実行が可能
• Tensor, Module, Optimizer, Loss関数などPyTorchの主要な機能を搭載
• ATen / Autograd / C++ Frontend / C++ Extensions
• C++なのでモバイルなどへの汎用性やマルチコアによる並列処理
トピック
• 専用フォルダをダウンロード ⇒ CMakeで簡単にセットアップ
• PyTorch Mobileでも内部ではLibTorchが動作
• Java APIもある。Android開発ではこちらを使うことになる
• pybind11によるPythonバインドやTorchScriptによる変換などでPython環境への配置もサポート
LibTorchとは