Use math.prod

tensorly · Jun 7, 2024 · 6b93e33 · 6b93e33
1 parent 14eae21
commit 6b93e33
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tltorch/functional/tests/test_factorized_linear.py b/tltorch/functional/tests/test_factorized_linear.py
@@ -5,7 +5,7 @@
 import tensorly as tl
 tl.set_backend('pytorch')
 from tensorly import testing
-from tensorly.utils import prod
+from math import prod
 
 import pytest
 
@@ -30,4 +30,4 @@ def test_linear_tensor_dot_tucker(factorization, factorized_linear):
     res = factorized_linear(tensor, fact_weight, transpose=True)
     res = res.reshape(batch_size, -1)
     testing.assert_array_almost_equal(true_res, res, decimal=5)
-
+