Add the slice indexing.

This commit is contained in:
laurent
2023-06-20 10:50:58 +01:00
parent 786544292d
commit 6c5fc767a8
3 changed files with 81 additions and 4 deletions

View File

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