Add graph canonization

- Add overloading of @ for Python matrix multiplication
benruijl · Aug 23, 2024 · b62d669 · b62d669
1 parent bd0536c
commit b62d669
Show file tree

Hide file tree

Showing 4 changed files with 551 additions and 4 deletions.
diff --git a/src/api/python.rs b/src/api/python.rs
@@ -9022,7 +9022,7 @@ impl PythonMatrix {
         self.__add__(rhs.__neg__())
     }
 
-    /// Add this matrix to `rhs`, returning the result.
+    /// Matrix multiply `self` and `rhs`, returning the result.
     pub fn __mul__(&self, rhs: ScalarOrMatrix) -> PyResult<PythonMatrix> {
         match rhs {
             ScalarOrMatrix::Scalar(s) => {
@@ -9041,11 +9041,21 @@ impl PythonMatrix {
         }
     }
 
-    /// Add this matrix to `rhs`, returning the result.
+    /// Matrix multiply `rhs` and `self` returning the result.
     pub fn __rmul__(&self, rhs: ConvertibleToRationalPolynomial) -> PyResult<PythonMatrix> {
         self.__mul__(ScalarOrMatrix::Scalar(rhs))
     }
 
+    /// Matrix multiply this matrix and `self`, returning the result.
+    pub fn __matmul__(&self, rhs: ScalarOrMatrix) -> PyResult<PythonMatrix> {
+        self.__mul__(rhs)
+    }
+
+    /// Matrix multiply `rhs` and `self`, returning the result.
+    pub fn __rmatmul__(&self, rhs: ConvertibleToRationalPolynomial) -> PyResult<PythonMatrix> {
+        self.__mul__(ScalarOrMatrix::Scalar(rhs))
+    }
+
     /// Divide the matrix by the scalar, returning the result.
     pub fn __truediv__(&self, rhs: ConvertibleToRationalPolynomial) -> PyResult<PythonMatrix> {
         Ok(PythonMatrix {