Use math.prod

Author: JeanKossaifi

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)
-    
+