1 files changed, 2 insertions, 2 deletions
diff --git a/candle-core/examples/cuda_basics.rs b/candle-core/examples/cuda_basics.rs
index aeee541a..37a66cb5 100644
--- a/candle-core/examples/cuda_basics.rs
+++ b/candle-core/examples/cuda_basics.rs
@@ -7,9 +7,9 @@ use candle::{Device, Tensor};
 fn main() -> Result<()> {
     let device = Device::new_cuda(0)?;
     let t = Tensor::new(&[[1f32, 2., 3., 4.2]], &device)?;
-    let sum = t.sum(&[0])?;
+    let sum = t.sum_keepdim(&[0])?;
     println!("{sum}");
-    let sum = t.sum(&[1])?;
+    let sum = t.sum_keepdim(&[1])?;
     println!("{sum}");
     Ok(())
 }