mirror of
https://github.com/huggingface/candle.git
synced 2025-06-16 18:48:51 +00:00
Add the elu op. (#113)
This commit is contained in:
@ -46,6 +46,7 @@ pub(crate) enum Op {
|
||||
Transpose(Tensor, usize, usize),
|
||||
Gelu(Tensor),
|
||||
Relu(Tensor),
|
||||
Elu(Tensor, f64),
|
||||
// TODO: Support for custom ops.
|
||||
}
|
||||
|
||||
|
Reference in New Issue
Block a user