回答編集履歴
2
d
test
CHANGED
|
@@ -2,7 +2,7 @@
|
|
|
2
2
|
|
|
3
3
|
|
|
4
4
|
|
|
5
|
-
演算するときに一方のテンソルのデバイスが GPU で、もう一方のテンソルが CPU
|
|
5
|
+
演算するときに一方のテンソルのデバイスが GPU で、もう一方のテンソルのデバイスが CPU になっているときに起こるエラーです。
|
|
6
6
|
|
|
7
7
|
`Expected object of type torch.cuda.FloatTensor` と言われているので、演算前に型が `torch.FloatTensor` のテンソルを Tensor.to("cuda") で `torch.cuda.FloatTensor` に変更すればいいと思います。
|
|
8
8
|
|
1
d
test
CHANGED
|
@@ -28,4 +28,6 @@
|
|
|
28
28
|
|
|
29
29
|
|
|
30
30
|
|
|
31
|
+
参考
|
|
32
|
+
|
|
31
33
|
[Pytorch - 計算を行うデバイスを指定する方法について](https://pystyle.info/pytorch-how-to-specify-the-device-for-calculation/)
|