Revision of Burn-readme-1.rs

1 file changed, 4 insertions

Burn-readme-1.rs(file created)

		@@ -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	+	}

更新更早