[tensorflow-sys] use product instead of fold

tensorflow · Nov 11, 2021 · aa9f6cc · aa9f6cc
1 parent 8e9b2a5
commit aa9f6cc
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tensorflow-sys/tests/lib.rs b/tensorflow-sys/tests/lib.rs
@@ -16,7 +16,7 @@ fn linkage() {
 fn tfe_tensor_handle() {
     let data = vec![0.0f32; 100];
     let shape = [1, 10, 10];
-    let num_elements = shape.iter().fold(1, |t, x| t * x) as usize;
+    let num_elements = shape.iter().product::<i64>() as usize;
 
     unsafe {
         let tf_tensor = ffi::TF_AllocateTensor(