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

Bugfix for the contiguous strides.

Browse Source
This commit is contained in:
laurent
2023-06-20 13:35:07 +01:00
parent d922ff97f2
commit 98b423145a
5 changed files with 64 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
tests/tensor_tests.rs
View File

@ -11,7 +11,7 @@ fn zeros() -> Result<()> {
#[test]
fn add_mul() -> Result<()> {
let tensor = Tensor::new([3f32, 1., 4.].as_slice(), Device::Cpu)?;
let tensor = Tensor::new(&[3f32, 1., 4.], Device::Cpu)?;
let dim1 = tensor.shape().r1()?;
assert_eq!(dim1, 3);
let content: Vec<f32> = tensor.to_vec1()?;
@ -24,3 +24,14 @@ fn add_mul() -> Result<()> {
assert_eq!(content, [36., 4., 64.]);
Ok(())
}
#[test]
fn tensor_2d() -> Result<()> {
let data = &[[3f32, 1., 4., 1., 5.], [2., 1., 7., 8., 2.]];
let tensor = Tensor::new(data, Device::Cpu)?;
let dims = tensor.shape().r2()?;
assert_eq!(dims, (2, 5));
let content: Vec<Vec<f32>> = tensor.to_vec2()?;
assert_eq!(content, data);
Ok(())
}