diff options
Diffstat (limited to 'candle-core/examples/cuda_basics.rs')
| -rw-r--r-- | candle-core/examples/cuda_basics.rs | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/candle-core/examples/cuda_basics.rs b/candle-core/examples/cuda_basics.rs index 12febb60..ac435488 100644 --- a/candle-core/examples/cuda_basics.rs +++ b/candle-core/examples/cuda_basics.rs @@ -11,7 +11,7 @@ fn main() -> Result<()> { let device = Device::new_cuda(0)?; let t = Tensor::randn(0f32, 1f32, (2, 4, 96, 96), &device)?; let w = Tensor::randn(0f32, 1f32, (320, 4, 3, 3), &device)?; - let res = t.conv2d(&w, 1, 1)?; + let res = t.conv2d(&w, 1, 1, 1)?; println!("{res:?}"); Ok(()) } |