mirror of
https://github.com/huggingface/candle.git
synced 2025-06-16 02:38:10 +00:00
Another prelu bugfix. (#1407)
This commit is contained in:
@ -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()
|
||||
};
|
||||
|
Reference in New Issue
Block a user