Q&A
オートエンコーダを実装する際に以下のようなエラーが発生してしまいます、解決策がわかる方がいましたらご教授お願いいたします。
Tensor for 'out' is on CPU, Tensor for argument #1 'self' is on CPU, but expected them to be on GPU (while checking arguments for addmm)
回答1件
0
ベストアンサー
このコードだけではわかりません。
Tensorの演算で使用しているtorch.Tensor型のオブジェクトがtorch.cuda.FloatTensorとかtorch.cuda.IntTensorのようなGPU tensorでなければならないときにtorch.FloatTensorとかtorch.IntTensorのようなCPU tensorを使っているようです。
コード全体を調べてtorch.Tensor型のオブジェクトの定義で間違っているところを修正する必要があります。
参考
投稿2021/12/19 22:34
総合スコア24641
あなたの回答
tips
プレビュー
下記のような回答は推奨されていません。
このような回答には修正を依頼しましょう。