diff options
| author | Laurent Mazare <laurent.mazare@gmail.com> | 2023-12-06 09:54:41 +0100 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-12-06 09:54:41 +0100 |
| commit | 236b820e286fcdc01ad7923125d9dac1eaa9c18d (patch) | |
| tree | 8adb292b2d7358b60ae932d21c0b14537fc1c2c1 /candle-nn | |
| parent | 2648e797c22ea9a25dd43cf6ecf0222dfd221170 (diff) | |
| download | candle-236b820e286fcdc01ad7923125d9dac1eaa9c18d.tar.gz candle-236b820e286fcdc01ad7923125d9dac1eaa9c18d.tar.bz2 candle-236b820e286fcdc01ad7923125d9dac1eaa9c18d.zip | |
Another prelu bugfix. (#1407)
Diffstat (limited to 'candle-nn')
| -rw-r--r-- | candle-nn/src/activation.rs | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/candle-nn/src/activation.rs b/candle-nn/src/activation.rs index c7dd359f..80b750ed 100644 --- a/candle-nn/src/activation.rs +++ b/candle-nn/src/activation.rs @@ -73,7 +73,7 @@ impl candle::Module for PReLU { } let mut s = vec![1; xs.rank()]; s[1] = self.weight.elem_count(); - self.weight.broadcast_as(s)? + self.weight.reshape(s)? } else { self.weight.clone() }; |