From d40a2d871e4b6220e9abd3e5e8065841ca168afa Mon Sep 17 00:00:00 2001
From: Ma Jian <jianma@habana.ai>
Date: Tue, 24 Dec 2024 09:51:01 +0200
Subject: [PATCH] fix basics example

---
 candle-core/examples/basics.rs | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/candle-core/examples/basics.rs b/candle-core/examples/basics.rs
index fe15187b5a..20581a89c8 100644
--- a/candle-core/examples/basics.rs
+++ b/candle-core/examples/basics.rs
@@ -9,9 +9,9 @@ use candle_core::{Device, Tensor};
 
 fn main() -> Result<()> {
     let a = Tensor::new(&[[0.0f32, 1.0, 2.0], [3.0, 4.0, 5.0]], &Device::Cpu)?;
-    let b = Tensor::new(&[[88.0f32, 99.0]], &Device::Cpu)?;
+    let b = Tensor::new(&[[88.0f32], [99.0]], &Device::Cpu)?;
     let new_a = a.slice_scatter(&b, 1, 2)?;
     assert_eq!(a.to_vec2::<f32>()?, [[0.0, 1.0, 2.0], [3.0, 4.0, 5.0]]);
-    assert_eq!(new_a.to_vec2::<f32>()?, [[0.0, 1.0, 2.0], [3.0, 4.0, 5.0]]);
+    assert_eq!(new_a.to_vec2::<f32>()?, [[0.0, 1.0, 88.0], [3.0, 4.0, 99.0]]);
     Ok(())
 }