Add diagonal() for Paddle

meyer-lab · Aug 12, 2024 · df19c2b · df19c2b
1 parent d616a32
commit df19c2b
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tensorly/backend/paddle_backend.py b/tensorly/backend/paddle_backend.py
@@ -293,6 +293,7 @@ def sign(tensor: paddle.Tensor):
         "stack",
         "logsumexp",
         "max",
+        "diagonal",
     ]
 ):
     if name in ["pi", "e", "inf", "nan"]: