chore: import upstream snapshot with attribution

2026-07-13 13:30:25 +08:00
commit f19b2512d7
562 changed files with 38082 additions and 0 deletions
@@ -0,0 +1,36 @@
+import numpy as np
+from prml.nn.tensor.constant import Constant
+from prml.nn.tensor.tensor import Tensor
+from prml.nn.function import Function
+
+
+class Transpose(Function):
+
+    def __init__(self, axes=None):
+        self.axes = axes
+
+    def forward(self, x):
+        x = self._convert2tensor(x)
+        if self.axes is not None:
+            self._equal_ndim(x, len(self.axes))
+        self.x = x
+        if isinstance(self.x, Constant):
+            return Constant(np.transpose(x.value, self.axes))
+        return Tensor(np.transpose(x.value, self.axes), function=self)
+
+    def backward(self, delta):
+        if self.axes is None:
+            dx = np.transpose(delta)
+        else:
+            dx = np.transpose(delta, np.argsort(self.axes))
+        self.x.backward(dx)
+
+
+def transpose(x, axes=None):
+    return Transpose(axes).forward(x)
+
+
+def transpose_method(x, *args):
+    if args == ():
+        args = None
+    return Transpose(args).forward(x)