Add the elu op. (#113)

This commit is contained in:
Laurent Mazare
2023-07-09 21:56:31 +01:00
committed by GitHub
parent ea5dfa69bc
commit 270997a055
8 changed files with 95 additions and 0 deletions

View File

@ -46,6 +46,7 @@ pub(crate) enum Op {
Transpose(Tensor, usize, usize),
Gelu(Tensor),
Relu(Tensor),
Elu(Tensor, f64),
// TODO: Support for custom ops.
}