Fix TensorStack of sparse matrices

tum-pbs · Dec 16, 2024 · cd20034 · cd20034
1 parent 8a7b20b
commit cd20034
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/phiml/math/_tensors.py b/phiml/math/_tensors.py
@@ -1424,7 +1424,10 @@ def _is_tracer(self) -> bool:
 
     @property
     def requires_broadcast(self):
-        return self._varying_shapes or not self._shape.well_defined or self._is_tracer or self._tensors[0].shape.is_non_uniform
+        if self._varying_shapes or not self._shape.well_defined or self._is_tracer or self._tensors[0].shape.is_non_uniform:
+            return True
+        from ._sparse import is_sparse
+        return is_sparse(self)
 
     @property
     def stack_dim(self):