すでにC/C++層をチェーンャーにプラグインする方法

Question

私はすでにCPUを使用しているC++レイヤーを作成しました。チェーンャーフレームワークをプラグインしたいのですが、どうしたらいいですか？ CPUとGPUのレイヤーを1つにまとめることはできますか？

Answer 1

Cython、pybind11などのツールを使用してPythonからC++コードを呼び出して、C++レイヤをChainerに埋め込むことができます。あなたはそれを行うために少しのグルーコードを書かなければなりません（例えばNumPy配列バッファをC++で書かれたレイヤーで使われているデータフォーマットから/に変換し、あなたのレイヤーをChainerスタイルの関数に変換する;小さなPythonクラスを書くことによって）。

CPUとGPUを順方向/逆方向に計算するために、F.copy()を使用できます。バックプロックをサポートしています（https://docs.chainer.org/en/stable/reference/generated/chainer.functions.copy.html?highlight=copy参照）。