knox a révisé ce gist . Aller à la révision
1 file changed, 4 insertions
Burn-readme-1.rs(fichier créé)
| @@ -0,0 +1,4 @@ | |||
| 1 | + | fn gelu_custom<B: Backend, const D: usize>(x: Tensor<B, D>) -> Tensor<B, D> { | |
| 2 | + | let x = x.clone() * ((x / SQRT_2).erf() + 1); | |
| 3 | + | x / 2 | |
| 4 | + | } | |
Plus récent
Plus ancien