huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-21 04:10:46 +00:00
Code Issues Packages Projects Releases Wiki Activity

Start adding some ops.

Browse Source
This commit is contained in:
laurent
2023-06-20 08:41:19 +01:00
parent ef6760117f
commit 7a31ba93e4
5 changed files with 83 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tests/tensor_tests.rs
View File

@ -6,7 +6,7 @@ fn add() -> Result<()> {
let (dim1, dim2) = tensor.shape().r2()?;
assert_eq!(dim1, 5);
assert_eq!(dim2, 2);
let tensor = Tensor::new([3., 1., 4.].as_slice(), Device::Cpu)?;
let tensor = Tensor::new([3f32, 1., 4.].as_slice(), Device::Cpu)?;
let dim1 = tensor.shape().r1()?;
assert_eq!(dim1, 3);
Ok(())