diff options
| author | Laurent Mazare <laurent.mazare@gmail.com> | 2024-02-18 22:51:20 +0100 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2024-02-18 22:51:20 +0100 |
| commit | 3ba37443e5e2bd8a31b84ba5335f3805581a631e (patch) | |
| tree | c65fde7631bba256e5bb39453eec7d0d736a7bfb /candle-nn | |
| parent | 1fb728772d603e2ca5195eb8b123d5fc77c62fed (diff) | |
| download | candle-3ba37443e5e2bd8a31b84ba5335f3805581a631e.tar.gz candle-3ba37443e5e2bd8a31b84ba5335f3805581a631e.tar.bz2 candle-3ba37443e5e2bd8a31b84ba5335f3805581a631e.zip | |
Bugfix for applying the bias in conv1d-transpose. (#1732)
Diffstat (limited to 'candle-nn')
| -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)?) } } |