huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-19 03:54:56 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add the erf function. (#917)

Browse Source
This commit is contained in:
Laurent Mazare
2023-09-21 06:19:10 +01:00
committed by GitHub
parent ab1d40ea97
commit 7b26e513f1
4 changed files with 45 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
candle-core/tests/tensor_tests.rs
View File

@ -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(())
}