Class BatchNorm2D
java.lang.Object
org.apache.wayang.basic.model.op.Op
org.apache.wayang.basic.model.op.nn.BatchNorm
org.apache.wayang.basic.model.op.nn.BatchNorm2D
- All Implemented Interfaces:
Serializable
- See Also:
-
Nested Class Summary
-
Field Summary
Fields inherited from class org.apache.wayang.basic.model.op.nn.BatchNorm
epsilon, momentum, numFeatures
-
Constructor Summary
ConstructorsConstructorDescriptionBatchNorm2D
(int numFeatures) BatchNorm2D
(int numFeatures, float epsilon, float momentum) BatchNorm2D
(int numFeatures, float epsilon, float momentum, String name) BatchNorm2D
(int numFeatures, float epsilon, float momentum, String name, Op.DType dType) BatchNorm2D
(int numFeatures, float epsilon, float momentum, Op.DType dType) BatchNorm2D
(int numFeatures, String name) BatchNorm2D
(int numFeatures, String name, Op.DType dType) BatchNorm2D
(int numFeatures, Op.DType dType) -
Method Summary
Methods inherited from class org.apache.wayang.basic.model.op.nn.BatchNorm
getEpsilon, getMomentum, getNumFeatures, inputsRequired
-
Constructor Details
-
BatchNorm2D
public BatchNorm2D(int numFeatures) -
BatchNorm2D
-
BatchNorm2D
-
BatchNorm2D
-
BatchNorm2D
public BatchNorm2D(int numFeatures, float epsilon, float momentum) -
BatchNorm2D
-
BatchNorm2D
-
BatchNorm2D
-