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.rs8
1 files changed, 8 insertions, 0 deletions
diff --git a/candle-core/tests/tensor_tests.rs b/candle-core/tests/tensor_tests.rs
index ae1bd058..899efcf3 100644
--- a/candle-core/tests/tensor_tests.rs
+++ b/candle-core/tests/tensor_tests.rs
@@ -1089,3 +1089,11 @@ fn pad_with_same() -> Result<()> {
);
Ok(())
}
+
+#[test]
+fn i64_abs() -> Result<()> {
+ let t = Tensor::new(&[-42i64, 1337], &Device::Cpu)?;
+ let t = t.abs()?;
+ assert_eq!(t.to_vec1::<i64>()?, [42, 1337]);
+ Ok(())
+}