From aba1e90797e430f28eec13b14b76dd5355876f9c Mon Sep 17 00:00:00 2001 From: Laurent Mazare Date: Wed, 23 Aug 2023 12:58:55 +0100 Subject: Add some group parameter to convolutions. (#566) * Add some group parameter to convolutions. * Avoid some unnecessary groups checks. * Move the tensor convolution bits. * Properh handling of groups. * Bump the crate version. * And add a changelog. --- candle-core/examples/basics.rs | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'candle-core/examples/basics.rs') diff --git a/candle-core/examples/basics.rs b/candle-core/examples/basics.rs index efce913a..9d4734de 100644 --- a/candle-core/examples/basics.rs +++ b/candle-core/examples/basics.rs @@ -11,7 +11,7 @@ fn main() -> Result<()> { let inp = Tensor::randn(0f32, 1., (2, 320, 96, 96), &Device::Cpu)?; let w = Tensor::randn(0f32, 1., (320, 320, 3, 3), &Device::Cpu)?; let start = std::time::Instant::now(); - let res = inp.conv2d(&w, 0, 1); + let res = inp.conv2d(&w, 0, 1, 1)?; println!("{:?}", start.elapsed()); println!("{res:?}"); Ok(()) -- cgit v1.2.3