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 d938cae4..8fbe7659 100644 --- a/candle-nn/src/conv.rs +++ b/candle-nn/src/conv.rs @@ -41,7 +41,7 @@ impl Conv1d { match &self.bias { None => Ok(x), Some(bias) => { - let b = bias.shape().r1()?; + let b = bias.dims1()?; let bias = bias.reshape((1, b, 1))?; Ok(x.broadcast_add(&bias)?) } |