summaryrefslogtreecommitdiff
path: root/candle-core/tests/tensor_tests.rs
diff options
context:
space:
mode:
Diffstat (limited to 'candle-core/tests/tensor_tests.rs')
-rw-r--r--candle-core/tests/tensor_tests.rs7
1 files changed, 7 insertions, 0 deletions
diff --git a/candle-core/tests/tensor_tests.rs b/candle-core/tests/tensor_tests.rs
index 408f4c55..edd0bd79 100644
--- a/candle-core/tests/tensor_tests.rs
+++ b/candle-core/tests/tensor_tests.rs
@@ -61,6 +61,13 @@ fn unary_op(device: &Device) -> Result<()> {
[2.6906, -0.0647, -0.1091, 1.7353, 2.7928]
]
);
+ assert_eq!(
+ test_utils::to_vec2_round(&tensor.erf()?, 4)?,
+ [
+ [-1.0, 0.8427, 1.0, -0.1125, 0.5205],
+ [0.9999, -0.9891, -0.3079, 0.9891, 0.9999]
+ ]
+ );
Ok(())
}