huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-17 11:08:52 +00:00
Code Issues Packages Projects Releases Wiki Activity

Shuffle the shape bits around.

Browse Source
This commit is contained in:
laurent
2023-06-19 20:22:12 +01:00
parent 634e0c88ae
commit 01eeb0e72f
5 changed files with 159 additions and 63 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tests/tensor_tests.rs
View File

@ -3,7 +3,7 @@ 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()?;
let (dim1, dim2) = tensor.shape().r2()?;
assert_eq!(dim1, 5);
assert_eq!(dim2, 2);
Ok(())