From 3ba37443e5e2bd8a31b84ba5335f3805581a631e Mon Sep 17 00:00:00 2001
From: Laurent Mazare <laurent.mazare@gmail.com>
Date: Sun, 18 Feb 2024 22:51:20 +0100
Subject: [PATCH] Bugfix for applying the bias in conv1d-transpose. (#1732)

---
 candle-nn/src/conv.rs | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/candle-nn/src/conv.rs b/candle-nn/src/conv.rs
index 03b69bbd26..c183e6b9f9 100644
--- a/candle-nn/src/conv.rs
+++ b/candle-nn/src/conv.rs
@@ -134,7 +134,7 @@ impl crate::Module for ConvTranspose1d {
             None => Ok(x),
             Some(bias) => {
                 let b = bias.dims1()?;
-                let bias = bias.reshape((1, b, 1, 1))?;
+                let bias = bias.reshape((1, b, 1))?;
                 Ok(x.broadcast_add(&bias)?)
             }
         }