mirror of
https://github.com/huggingface/candle.git
synced 2025-06-16 10:38:54 +00:00
85 lines
2.7 KiB
Rust
85 lines
2.7 KiB
Rust
use anyhow::Result;
|
|
use candle::{DType, Device::Cpu, Tensor};
|
|
|
|
#[test]
|
|
fn display_scalar() -> Result<()> {
|
|
let t = Tensor::new(1234u32, &Cpu)?;
|
|
let s = format!("{t}");
|
|
assert_eq!(&s, "[1234]\nTensor[[], u32]");
|
|
let t = t.to_dtype(DType::F32)?.neg()?;
|
|
let s = format!("{}", (&t / 10.0)?);
|
|
assert_eq!(&s, "[-123.4000]\nTensor[[], f32]");
|
|
let s = format!("{}", (&t / 1e8)?);
|
|
assert_eq!(&s, "[-1.2340e-5]\nTensor[[], f32]");
|
|
let s = format!("{}", (&t * 1e8)?);
|
|
assert_eq!(&s, "[-1.2340e11]\nTensor[[], f32]");
|
|
let s = format!("{}", (&t * 0.)?);
|
|
assert_eq!(&s, "[0.]\nTensor[[], f32]");
|
|
Ok(())
|
|
}
|
|
|
|
#[test]
|
|
fn display_vector() -> Result<()> {
|
|
let t = Tensor::new::<&[u32; 0]>(&[], &Cpu)?;
|
|
let s = format!("{t}");
|
|
assert_eq!(&s, "[]\nTensor[[0], u32]");
|
|
let t = Tensor::new(&[0.1234567, 1.0, -1.2, 4.1, f64::NAN], &Cpu)?;
|
|
let s = format!("{t}");
|
|
assert_eq!(
|
|
&s,
|
|
"[ 0.1235, 1.0000, -1.2000, 4.1000, NaN]\nTensor[[5], f64]"
|
|
);
|
|
let t = (Tensor::ones(50, DType::F32, &Cpu)? * 42.)?;
|
|
let s = format!("\n{t}");
|
|
let expected = r#"
|
|
[42., 42., 42., 42., 42., 42., 42., 42., 42., 42., 42., 42., 42., 42., 42., 42.,
|
|
42., 42., 42., 42., 42., 42., 42., 42., 42., 42., 42., 42., 42., 42., 42., 42.,
|
|
42., 42., 42., 42., 42., 42., 42., 42., 42., 42., 42., 42., 42., 42., 42., 42.,
|
|
42., 42.]
|
|
Tensor[[50], f32]"#;
|
|
assert_eq!(&s, expected);
|
|
let t = (Tensor::ones(11000, DType::F32, &Cpu)? * 42.)?;
|
|
let s = format!("{t}");
|
|
assert_eq!(
|
|
&s,
|
|
"[42., 42., 42., ..., 42., 42., 42.]\nTensor[[11000], f32]"
|
|
);
|
|
Ok(())
|
|
}
|
|
|
|
#[test]
|
|
fn display_multi_dim() -> Result<()> {
|
|
let t = (Tensor::ones((200, 100), DType::F32, &Cpu)? * 42.)?;
|
|
let s = format!("\n{t}");
|
|
let expected = r#"
|
|
[[42., 42., 42., ..., 42., 42., 42.],
|
|
[42., 42., 42., ..., 42., 42., 42.],
|
|
[42., 42., 42., ..., 42., 42., 42.],
|
|
...
|
|
[42., 42., 42., ..., 42., 42., 42.],
|
|
[42., 42., 42., ..., 42., 42., 42.],
|
|
[42., 42., 42., ..., 42., 42., 42.]]
|
|
Tensor[[200, 100], f32]"#;
|
|
assert_eq!(&s, expected);
|
|
let t = t.reshape(&[2, 1, 1, 100, 100])?;
|
|
let t = format!("\n{t}");
|
|
let expected = r#"
|
|
[[[[[42., 42., 42., ..., 42., 42., 42.],
|
|
[42., 42., 42., ..., 42., 42., 42.],
|
|
[42., 42., 42., ..., 42., 42., 42.],
|
|
...
|
|
[42., 42., 42., ..., 42., 42., 42.],
|
|
[42., 42., 42., ..., 42., 42., 42.],
|
|
[42., 42., 42., ..., 42., 42., 42.]]]],
|
|
[[[[42., 42., 42., ..., 42., 42., 42.],
|
|
[42., 42., 42., ..., 42., 42., 42.],
|
|
[42., 42., 42., ..., 42., 42., 42.],
|
|
...
|
|
[42., 42., 42., ..., 42., 42., 42.],
|
|
[42., 42., 42., ..., 42., 42., 42.],
|
|
[42., 42., 42., ..., 42., 42., 42.]]]]]
|
|
Tensor[[2, 1, 1, 100, 100], f32]"#;
|
|
assert_eq!(&t, expected);
|
|
Ok(())
|
|
}
|