release code

Co-Authored-By: Claude Opus 4.7 <noreply@anthropic.com>
2026-10-12 22:30:08 +08:00
commit 5027694a2b
291 changed files with 60190 additions and 0 deletions
@@ -0,0 +1,15 @@
+import torch.nn as nn
+
+
+class Sequential2(nn.Sequential):
+    """An alternative sequential container to nn.Sequential,
+    which accepts an arbitrary number of input arguments.
+    """
+
+    def forward(self, *inputs):
+        for module in self._modules.values():
+            if isinstance(inputs, tuple):
+                inputs = module(*inputs)
+            else:
+                inputs = module(inputs)
+        return inputs