1 files changed, 1 insertions, 1 deletions
diff --git a/candle-core/src/conv.rs b/candle-core/src/conv.rs
index 710f6abf..fe923087 100644
--- a/candle-core/src/conv.rs
+++ b/candle-core/src/conv.rs
@@ -196,8 +196,8 @@ impl Tensor {
         stride: usize,
         dilation: usize,
     ) -> Result<Self> {
-        let (c_out, c_in_k, k_size) = kernel.dims3()?;
         let (b_size, c_in, l_in) = self.dims3()?;
+        let (c_in_k, c_out, k_size) = kernel.dims3()?;
         if c_in != c_in_k {
             crate::bail!("in_channel mismatch between input ({c_in}) and kernel ({c_in_k})")
         }