refactor: update Jupyter notebook and main script for improved tensor operations and output display

2025-06-12 11:46:42 +08:00
parent 8bf116cbde
commit 97637b29a2
2 changed files with 211 additions and 18 deletions
--- a/main.py
+++ b/main.py
@ -1,4 +1,4 @@
 import torch
 import numpy as np

-torch.__version__
+print(torch.__version__)