Implement group-norm. (#334)

* Implement group-norm. * Add some testing for group-norm.
2025-06-17 02:58:50 +00:00 · 2023-08-07 07:53:05 +02:00
parent 2c9f605976
commit 5bb2fce998
5 changed files with 150 additions and 14 deletions
--- a/candle-examples/examples/stable-diffusion/utils.rs
+++ b/candle-examples/examples/stable-diffusion/utils.rs
@ -1,10 +1,5 @@
 use candle::{Device, Result, Tensor};

-pub fn sigmoid(xs: &Tensor) -> Result<Tensor> {
-    // TODO: Add sigmoid as binary ops.
-    (xs.neg()?.exp()? - 1.0)?.recip()
-}
-
 pub fn avg_pool2d(_: &Tensor) -> Result<Tensor> {
    todo!()
 }