Package deepboof.impl.forward.standard
Class FunctionBatchNorm_F64
java.lang.Object
deepboof.impl.forward.standard.BaseFunction<Tensor_F64>
deepboof.impl.forward.standard.FunctionBatchNorm_F64
- All Implemented Interfaces:
BatchNorm,FunctionBatchNorm<Tensor_F64>,Function<Tensor_F64>
- Direct Known Subclasses:
SpatialBatchNorm_F64
public class FunctionBatchNorm_F64 extends BaseFunction<Tensor_F64> implements FunctionBatchNorm<Tensor_F64>
Implementation of
FunctionBatchNorm for Tensor_F64.-
Field Summary
Fields Modifier and Type Field Description protected doubleEPSprotected Tensor_F64paramsprotected booleanrequiresGammaBetaFields inherited from class deepboof.impl.forward.standard.BaseFunction
miniBatchSize, parameters, shapeInput, shapeOutput, shapeParameters -
Constructor Summary
Constructors Constructor Description FunctionBatchNorm_F64(boolean requiresGammaBeta) -
Method Summary
Modifier and Type Method Description void_forward(Tensor_F64 input, Tensor_F64 output)void_initialize()void_setParameters(java.util.List<Tensor_F64> parameters)doublegetEPS()java.lang.Class<Tensor_F64>getTensorType()Returns the type of tensor it can processbooleanhasGammaBeta()If it returns true then it expects a second set of parameters that defines gamma and beta.voidsetEPS(double EPS)Used to specify the EPS value.Methods inherited from class deepboof.impl.forward.standard.BaseFunction
forward, getOutputShape, getParameters, getParameterShapes, initialize, setParametersMethods inherited from class java.lang.Object
clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, waitMethods inherited from interface deepboof.Function
getOutputShape, getParameters, getParameterShapes, initialize
-
Field Details
-
requiresGammaBeta
protected boolean requiresGammaBeta -
params
-
EPS
protected double EPS
-
-
Constructor Details
-
FunctionBatchNorm_F64
public FunctionBatchNorm_F64(boolean requiresGammaBeta)
-
-
Method Details
-
_initialize
public void _initialize()- Specified by:
_initializein classBaseFunction<Tensor_F64>
-
_setParameters
- Specified by:
_setParametersin classBaseFunction<Tensor_F64>
-
_forward
- Specified by:
_forwardin classBaseFunction<Tensor_F64>
-
getEPS
public double getEPS() -
setEPS
public void setEPS(double EPS)Description copied from interface:BatchNormUsed to specify the EPS value. Must be invoked before setParameters() -
hasGammaBeta
public boolean hasGammaBeta()Description copied from interface:BatchNormIf it returns true then it expects a second set of parameters that defines gamma and beta.- Specified by:
hasGammaBetain interfaceBatchNorm- Returns:
- true if gamma and beta is returned.
-
getTensorType
Description copied from interface:FunctionReturns the type of tensor it can process- Specified by:
getTensorTypein interfaceFunction<Tensor_F64>- Returns:
- Type of tensor
-