From a2f72edc0de31709f9489922f93bb52b50035ebd Mon Sep 17 00:00:00 2001
From: Laurent Mazare <laurent.mazare@gmail.com>
Date: Fri, 14 Jul 2023 08:22:08 +0100
Subject: Simplify the parameters used by sum and sum_keepdim. (#165)

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

(limited to 'candle-core/examples/cuda_basics.rs')

diff --git a/candle-core/examples/cuda_basics.rs b/candle-core/examples/cuda_basics.rs
index 37a66cb5..645420e5 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_keepdim(&[0])?;
+    let sum = t.sum_keepdim(0)?;
     println!("{sum}");
-    let sum = t.sum_keepdim(&[1])?;
+    let sum = t.sum_keepdim(1)?;
     println!("{sum}");
     Ok(())
 }
-- 
cgit v1.2.3