Add some basic test.

This commit is contained in:
laurent
2023-06-19 19:50:17 +01:00
parent 8e2c534d1f
commit 634e0c88ae
4 changed files with 84 additions and 1 deletions

10
tests/tensor_tests.rs Normal file
View File

@ -0,0 +1,10 @@
use candle::{DType, Device, Result, Tensor};
#[test]
fn add() -> Result<()> {
let tensor = Tensor::zeros(&[5, 2], DType::F32, Device::Cpu);
let (dim1, dim2) = tensor.shape2()?;
assert_eq!(dim1, 5);
assert_eq!(dim2, 2);
Ok(())
}