Merge pull request #3062 from davenpi/fix/core-basics-example

Fix broken slice_scatter example in basics.rs

Author: ivarflakstad

candle-core/examples/basics.rs

@@ -9,9 +9,12 @@ 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(())
 }