demo 1

app.py

@@ -17,10 +17,7 @@ RandomTensor.to(device)
 
 def sysinfo(newdev = device):
     currentDevice = RandomTensor.get_device();
-    tensorExample = RandomTensor.to(newdev)
-                                .norm(p=2, dim=1, keepdim=True) # Apply an operation similar to l2_norm
-                                .unsqueeze(-1)                 # Add an extra dimension
-                                .to(newdev) 
+    tensorExample = RandomTensor.to(newdev).norm(p=2, dim=1, keepdim=True).unsqueeze(-1).to(newdev) 
     
     tocpu = tensorExample.cpu().squeeze().half().tolist()