ai:pytorch

差分

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

この比較画面へのリンク

--- ai:pytorch [2019/12/29 17:05] – [メモ] oga
+++ ai:pytorch [2019/12/29 17:08] – [メモ] oga
@@ 行 36: / 行 36: @@
 <code cpp>
+#include  <torch/torch.h>
 class ModelFCImpl : public torch::nn::Module {
     torch::nn::Linear  fc1= nullptr;
@@ 行 59: / 行 61: @@
 </code>
+Shape 定義は NCHW 形式で IntArrayRef を使う。<nowiki>std::vector<int64_t></nowiki> が利用可能。
+TORCH_MODULE() で定義しておけば直接 torch::save() や torch::load() が使える。

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