Package org.apache.wayang.basic.model.op.nn

Class BatchNorm

java.lang.Object

org.apache.wayang.basic.model.op.Op

org.apache.wayang.basic.model.op.nn.BatchNorm

All Implemented Interfaces:

Serializable

Direct Known Subclasses:

BatchNorm2D, BatchNorm3D

public abstract class BatchNorm
extends Op

See Also:

• Serialized Form

Nested Class Summary

Nested classes/interfaces inherited from class org.apache.wayang.basic.model.op.Op

Op.DType

Field Summary

Fields

Modifier and Type	Field	Description
protected final float	epsilon
protected final float	momentum
protected final int	numFeatures

Fields inherited from class org.apache.wayang.basic.model.op.Op

dType, fromList, id, name

Constructor Summary

Constructors

Constructor	Description
BatchNorm(int numFeatures, float epsilon, float momentum, String name, Op.DType dType)
BatchNorm(int numFeatures, String name, Op.DType dType)

Method Summary

Modifier and Type	Method	Description
float	getEpsilon()
float	getMomentum()
int	getNumFeatures()
int	inputsRequired()

Methods inherited from class org.apache.wayang.basic.model.op.Op

getDType, getFromList, getId, getName, with

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Field Details

numFeatures

protected final int numFeatures

epsilon

protected final float epsilon

momentum

protected final float momentum

Constructor Details

BatchNorm

public BatchNorm(int numFeatures,
 String name,
 Op.DType dType)

BatchNorm

public BatchNorm(int numFeatures,
 float epsilon,
 float momentum,
 String name,
 Op.DType dType)

Method Details

getNumFeatures

public int getNumFeatures()

getEpsilon

public float getEpsilon()

getMomentum

public float getMomentum()

inputsRequired

public int inputsRequired()

Specified by:

inputsRequired in class Op