huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-17 11:08:52 +00:00
Code Issues Packages Projects Releases Wiki Activity

Cosmetic change to the backprop test.

Browse Source
This commit is contained in:
laurent
2023-06-20 20:56:50 +01:00
parent a419a9da72
commit e04d0ac1ac
1 changed files with 1 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
tests/grad_tests.rs
View File

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