diff options
Diffstat (limited to 'examples/basics.rs')
-rw-r--r-- | examples/basics.rs | 9 |
1 files changed, 9 insertions, 0 deletions
diff --git a/examples/basics.rs b/examples/basics.rs new file mode 100644 index 00000000..f01f7871 --- /dev/null +++ b/examples/basics.rs @@ -0,0 +1,9 @@ +use anyhow::Result; +use candle::{Device, Tensor}; + +fn main() -> Result<()> { + let x = Tensor::var(&[3f32, 1., 4.], Device::Cpu)?; + let y = (((&x * &x)? + &x * 5f64)? + 4f64)?; + println!("{:?}", y.to_vec1::<f32>()?); + Ok(()) +} |