generalize sigmoid transformation

bioimage-io · Sep 26, 2024 · 6abe193 · 6abe193
1 parent 0a23d1e
commit 6abe193
Showing 1 changed file with 11 additions and 4 deletions.
diff --git a/src/main/java/io/bioimage/modelrunner/transformations/SigmoidTransformation.java b/src/main/java/io/bioimage/modelrunner/transformations/SigmoidTransformation.java
@@ -32,17 +32,24 @@ public class SigmoidTransformation extends AbstractTensorPixelTransformation
 	public SigmoidTransformation()
 	{
 		super(name);
+		super.setFloatUnitaryOperator( v -> ( float ) ( 1. / ( 1. + Math.exp( -v ) ) ) );
+		super.setDoubleUnitaryOperator(v -> ( double ) ( 1. / ( 1. + Math.exp( -v ) ) ));
+		super.setByteUnitaryOperator(v -> ( byte ) ( 1. / ( 1. + Math.exp( -v ) ) ));
+		super.setUByteUnitaryOperator(v -> ( int ) ( 1. / ( 1. + Math.exp( -v ) ) ));
+		super.setShortUnitaryOperator(v -> ( short ) ( 1. / ( 1. + Math.exp( -v ) ) ));
+		super.setUShortUnitaryOperator(v -> ( int ) ( 1. / ( 1. + Math.exp( -v ) ) ));
+		super.setIntUnitaryOperator(v -> ( int ) ( 1. / ( 1. + Math.exp( -v ) ) ));
+		super.setUIntUnitaryOperator(v -> ( long ) ( 1. / ( 1. + Math.exp( -v ) ) ));
+		super.setLongUnitaryOperator(v -> ( long ) ( 1. / ( 1. + Math.exp( -v ) ) ));
 	}
 
 	public < R extends RealType< R > & NativeType< R > > Tensor< FloatType > apply( final Tensor< R > input )
 	{
-		super.setFloatUnitaryOperator( v -> ( float ) ( 1. / ( 1. + Math.exp( -v ) ) ) );
 		return super.apply(input);
 	}
 
-	public void applyInPlace( final Tensor< FloatType > input )
+	public < R extends RealType< R > & NativeType< R > > void applyInPlace( final Tensor< R > input )
 	{
-		super.setFloatUnitaryOperator( v -> ( float ) ( 1. / ( 1. + Math.exp( -v ) ) ) );
-		super.apply(input);
+		super.applyInPlace(input);
 	}
 }