回答編集履歴
2
d
answer
CHANGED
|
@@ -1,6 +1,6 @@
|
|
|
1
1
|
> RuntimeError: Expected object of type torch.cuda.FloatTensor but found type torch.FloatTensor for argument #2 'other'
|
|
2
2
|
|
|
3
|
-
演算するときに一方のテンソルのデバイスが GPU で、もう一方のテンソル
|
|
3
|
+
演算するときに一方のテンソルのデバイスが GPU で、もう一方のテンソルのデバイスが CPU になっているときに起こるエラーです。
|
|
4
4
|
`Expected object of type torch.cuda.FloatTensor` と言われているので、演算前に型が `torch.FloatTensor` のテンソルを Tensor.to("cuda") で `torch.cuda.FloatTensor` に変更すればいいと思います。
|
|
5
5
|
|
|
6
6
|
|
1
d
answer
CHANGED
|
@@ -13,4 +13,5 @@
|
|
|
13
13
|
c = a * b # エラー
|
|
14
14
|
```
|
|
15
15
|
|
|
16
|
+
参考
|
|
16
17
|
[Pytorch - 計算を行うデバイスを指定する方法について](https://pystyle.info/pytorch-how-to-specify-the-device-for-calculation/)
|