diff options
Diffstat (limited to 'candle-core/examples/basics.rs')
| -rw-r--r-- | candle-core/examples/basics.rs | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/candle-core/examples/basics.rs b/candle-core/examples/basics.rs index a5e2b24e..88b110e0 100644 --- a/candle-core/examples/basics.rs +++ b/candle-core/examples/basics.rs @@ -5,6 +5,11 @@ use anyhow::Result; use candle::{Device, Tensor}; fn main() -> Result<()> { + let a = Tensor::randn(0f32, 1., (2, 3), &Device::Cpu)?; + let b = Tensor::randn(0f32, 1., (3, 4), &Device::Cpu)?; + let c = a.matmul(&b)?; + println!("{a} {b} {c}"); + let data = &[[3f32, 1., 4., 1., 5.], [2., 7., 1., 8., 2.]]; let t1 = Tensor::new(data, &Device::Cpu)?; let data2 = &[[5f32, 5., 5., 5., 5.], [2., 7., 1., 8., 2.]]; |