Attempt fix for array API bug.

pydata · Nov 28, 2024 · 4456e41 · 4456e41
1 parent 3c87ca0
commit 4456e41
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/sparse/numba_backend/_common.py b/sparse/numba_backend/_common.py
@@ -157,6 +157,12 @@ def tensordot(a, b, axes=2, *, return_type=None):
     ndb = b.ndim
     equal = True
     if nda == 0 or ndb == 0:
+        if axes_a == [] and axes_b == []:
+            if nda == 0 and isinstance(a, SparseArray):
+                a = a.todense()
+            if ndb == 0 and isinstance(b, SparseArray):
+                b = b.todense()
+            return a * b
         pos = int(nda != 0)
         raise ValueError(f"Input {pos} operand does not have enough dimensions")
     if na != nb: