Skip to content

Commit 10d7af9

Browse files
committed
FIX: default batched_modes tensor_dot_tucker
1 parent f693ecd commit 10d7af9

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tltorch/functional/factorized_tensordot.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -42,7 +42,7 @@
4242
# return tl.einsum(eq, tensor, tucker.core, *tucker.factors)
4343

4444

45-
def tensor_dot_tucker(tensor, tucker, modes, batched_modes):
45+
def tensor_dot_tucker(tensor, tucker, modes, batched_modes=()):
4646
modes_tensor, modes_tucker = _validate_contraction_modes(
4747
tl.shape(tensor), tucker.tensor_shape, modes)
4848
input_order = tensor.ndim

0 commit comments

Comments
 (0)