From e04d0ac1ac2a233d9765419c8e46d5e249febecc Mon Sep 17 00:00:00 2001
From: laurent <laurent.mazare@gmail.com>
Date: Tue, 20 Jun 2023 20:56:50 +0100
Subject: [PATCH] Cosmetic change to the backprop test.

---
 tests/grad_tests.rs | 4 +---
 1 file changed, 1 insertion(+), 3 deletions(-)

diff --git a/tests/grad_tests.rs b/tests/grad_tests.rs
index d8149bea..4d7eae7d 100644
--- a/tests/grad_tests.rs
+++ b/tests/grad_tests.rs
@@ -5,9 +5,7 @@ use candle::{Device, Tensor};
 fn simple_grad() -> Result<()> {
     let x = Tensor::var(&[3f32, 1., 4.], Device::Cpu)?;
     let five = Tensor::new(&[5f32, 5., 5.], Device::Cpu)?;
-    let x_times_five = x.mul(&five)?;
-    let x2 = x.mul(&x)?;
-    let y = x2.add(&x_times_five)?.add(&five)?;
+    let y = x.mul(&x)?.add(&x.mul(&five)?)?.add(&five)?;
     let grads = y.backward()?;
     let grad_x = grads.get(&x.id()).context("no grad for x")?;
     assert_eq!(x.to_vec1::<f32>()?, [3., 1., 4.]);