Skip to content

Commit cb93e9f

Browse files
committed
Add NotImplementedError in einsum
1 parent f54c327 commit cb93e9f

File tree

1 file changed

+1
-0
lines changed
  • src/torchjd/sparse/_aten_function_overrides

1 file changed

+1
-0
lines changed

src/torchjd/sparse/_aten_function_overrides/einsum.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -11,6 +11,7 @@
1111

1212

1313
def einsum(*args: tuple[StructuredSparseTensor, list[int]], output: list[int]) -> Tensor:
14+
raise NotImplementedError()
1415

1516
# First part of the algorithm, determine how to cluster physical indices as well as the common
1617
# p_shapes corresponding to matching v_dims. Second part translates to physical einsum.

0 commit comments

Comments
 (0)