diff options
Diffstat (limited to 'candle-nn/src/conv.rs')
-rw-r--r-- | candle-nn/src/conv.rs | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/candle-nn/src/conv.rs b/candle-nn/src/conv.rs index 03b69bbd..c183e6b9 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)?) } } |