diff options
Diffstat (limited to 'examples/cuda_basics.rs')
| -rw-r--r-- | examples/cuda_basics.rs | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/examples/cuda_basics.rs b/examples/cuda_basics.rs index 046091a3..52a45999 100644 --- a/examples/cuda_basics.rs +++ b/examples/cuda_basics.rs @@ -5,9 +5,9 @@ fn main() -> Result<()> { let device = Device::new_cuda(0)?; let x = Tensor::new(&[3f32, 1., 4., 1., 5.], &device)?; println!("{:?}", x.to_vec1::<f32>()?); - let x = Tensor::new(&[2f32, 7., 1., 8., 2.], &device)?; - let y = (x * 3.)?; - println!("{:?}", y.to_vec1::<f32>()?); + let y = Tensor::new(&[2f32, 7., 1., 8., 2.], &device)?; + let z = (y + x * 3.)?; + println!("{:?}", z.to_vec1::<f32>()?); let x = Tensor::ones((3, 2), DType::F32, &device)?; println!("{:?}", x.to_vec2::<f32>()?); Ok(()) |