diff options
author | Laurent Mazare <laurent.mazare@gmail.com> | 2023-08-27 09:42:22 +0100 |
---|---|---|
committer | GitHub <noreply@github.com> | 2023-08-27 09:42:22 +0100 |
commit | 5320aa6b7d339ff594d3886dd29634ea8cde6f17 (patch) | |
tree | b8dc53eaea3966c288ac3c4a597b4d36c2deefa4 /candle-nn/tests/test_utils.rs | |
parent | a8b39dd7b784b3c3cdd4d228813bd48b2d0d79bb (diff) | |
download | candle-5320aa6b7d339ff594d3886dd29634ea8cde6f17.tar.gz candle-5320aa6b7d339ff594d3886dd29634ea8cde6f17.tar.bz2 candle-5320aa6b7d339ff594d3886dd29634ea8cde6f17.zip |
Move the test-utils bits to a shared place. (#619)
Diffstat (limited to 'candle-nn/tests/test_utils.rs')
-rw-r--r-- | candle-nn/tests/test_utils.rs | 39 |
1 files changed, 0 insertions, 39 deletions
diff --git a/candle-nn/tests/test_utils.rs b/candle-nn/tests/test_utils.rs deleted file mode 100644 index bb422cd9..00000000 --- a/candle-nn/tests/test_utils.rs +++ /dev/null @@ -1,39 +0,0 @@ -#![allow(dead_code)] -use candle::{Result, Tensor}; - -pub fn to_vec0_round(t: &Tensor, digits: i32) -> Result<f32> { - let b = 10f32.powi(digits); - let t = t.to_vec0::<f32>()?; - Ok(f32::round(t * b) / b) -} - -pub fn to_vec1_round(t: &Tensor, digits: i32) -> Result<Vec<f32>> { - let b = 10f32.powi(digits); - let t = t.to_vec1::<f32>()?; - let t = t.iter().map(|t| f32::round(t * b) / b).collect(); - Ok(t) -} - -pub fn to_vec2_round(t: &Tensor, digits: i32) -> Result<Vec<Vec<f32>>> { - let b = 10f32.powi(digits); - let t = t.to_vec2::<f32>()?; - let t = t - .iter() - .map(|t| t.iter().map(|t| f32::round(t * b) / b).collect()) - .collect(); - Ok(t) -} - -pub fn to_vec3_round(t: Tensor, digits: i32) -> Result<Vec<Vec<Vec<f32>>>> { - let b = 10f32.powi(digits); - let t = t.to_vec3::<f32>()?; - let t = t - .iter() - .map(|t| { - t.iter() - .map(|t| t.iter().map(|t| f32::round(t * b) / b).collect()) - .collect() - }) - .collect(); - Ok(t) -} |