Update pytorch_example.py

2023-11-09 20:10:53 +08:00
parent 8f4d27ffdb
commit 9260babdec
1 changed files with 1 additions and 2 deletions
--- a/2023.11.08_pytorch_example/pytorch_example.py
+++ b/2023.11.08_pytorch_example/pytorch_example.py
@@ -24,8 +24,7 @@ print('\n\n')
 print(int_tensor)
 print(float_tensor)
 print(double_tensor)
-
-
+print('张量在 CPU 或 GPU ：', int_tensor.device)


 # Torch数据和NumPy数据的转换