ai:pytorch

差分

このページの2つのバージョン間の差分を表示します。

この比較画面へのリンク

--- ai:pytorch [2019/12/29 17:05] – [メモ] oga
+++ ai:pytorch [2020/01/04 17:59] – [C++ API sample] oga
@@ 行 61: / 行 61: @@
 </code>
+Shape 定義は NCHW 形式で IntArrayRef を使う。<nowiki>std::vector<int64_t></nowiki> が利用可能。
+TORCH_MODULE() で定義しておけば直接 torch::save() や torch::load() が使える。
+===== C++ API sample =====
+https://github.com/hiroog/cppapimnist/tree/master/pytorch_cpp

ai/pytorch.txt · 最終更新: 2021/11/27 20:45 by oga