Update torch and bridge

Author: veehz

package.json

@@ -62,7 +62,7 @@
     "moo": "^0.5.2",
     "nearley": "^2.20.1",
     "pyodide": "^0.29.3",
-    "torch": "https://pkg.pr.new/veehz/torch@297abf6",
+    "torch": "https://pkg.pr.new/veehz/torch@687fc81",
     "wabt": "^1.0.37"
   }
 }

src/pyodide/bridge.py

@@ -184,8 +184,7 @@ def transpose(self, dim0, dim1):
         return Tensor(self._js.transpose(dim0, dim1))
 
     def flatten(self, start_dim=0, end_dim=-1):
-        n = self.numel()
-        return self.reshape([n])
+        return Tensor(self._js.flatten(start_dim, end_dim))
 
     # ------------------------------------------------------------------
     # Reductions — default (no dim) sums all elements, matching PyTorch
@@ -387,7 +386,7 @@ def __setattr__(self, name, value):
             object.__setattr__(self, name, value)
             return
 
-        if isinstance(value, Tensor) and value.requires_grad:
+        if isinstance(value, Parameter):
             params[name] = value
         elif isinstance(value, (Module, _NNModule)):
             modules[name] = value
@@ -674,17 +673,15 @@ def is_grad_enabled(self):
         return bool(js_torch.is_grad_enabled())
 
     def cat(self, tensors, dim=0):
-        """Concatenate tensors along dim. NOTE: gradient is not tracked."""
-        if dim != 0:
-            raise NotImplementedError("torch.cat only supports dim=0 in this bridge")
-        result = []
-        for t in tensors:
-            data = t.tolist()
-            if isinstance(data, list):
-                result.extend(data)
-            else:
-                result.append(data)
-        return Tensor(result)
+        if isinstance(tensors, Tensor):
+            tensors = [tensors]
+        return Tensor(js_torch.cat(to_js([t._js for t in tensors]), dim))
+
+    def concatenate(self, tensors, dim=0):
+        return self.cat(tensors, dim)
+
+    def concat(self, tensors, dim=0):
+        return self.cat(tensors, dim)
 
     def Size(self, shape):
         return list(shape)

yarn.lock

@@ -4306,7 +4306,7 @@ __metadata:
     prettier: "npm:^3.8.1"
     pyodide: "npm:^0.29.3"
     rollup: "npm:^4.59.0"
-    torch: "https://pkg.pr.new/veehz/torch@297abf6"
+    torch: "https://pkg.pr.new/veehz/torch@687fc81"
     ts-jest: "npm:^29.0.5"
     ts-node: "npm:^10.9.1"
     tslib: "npm:^2.8.1"
@@ -4818,10 +4818,10 @@ __metadata:
   languageName: node
   linkType: hard
 
-"torch@https://pkg.pr.new/veehz/torch@297abf6":
+"torch@https://pkg.pr.new/veehz/torch@687fc81":
   version: 0.1.0
-  resolution: "torch@https://pkg.pr.new/veehz/torch@297abf6"
-  checksum: 10c0/bde381a0d266c845970fd34b3d27048709b60fd9e64a2d96f3080eb9c3a4bb1433bcc0ea1bc1123fbb31883ad2623c031ede94319e514f54398b9752c290c6d4
+  resolution: "torch@https://pkg.pr.new/veehz/torch@687fc81"
+  checksum: 10c0/84fd0d0fae8f5e67dccfa01d429755ef5db147340544f697a14ce32332a31b5515b84aa47ac849c327f1026c8c4ff29a7af3c60bd5c1db541f49ddfa9f529185
   languageName: node
   linkType: hard