knox revidoval tento gist . Přejít na revizi
1 file changed, 4 insertions
Burn-readme-1.rs(vytvořil soubor)
| @@ -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 | + | } | |
Novější
Starší