com.intel.analytics.zoo.pipeline.api.keras.layers

BatchNormalization

class BatchNormalization[T] extends bigdl.nn.keras.BatchNormalization[T] with Net

Batch normalization layer. Normalize the activations of the previous layer at each batch, i.e. applies a transformation that maintains the mean activation close to 0 and the activation standard deviation close to 1. It is a feature-wise normalization, each feature map in the input will be normalized separately. The input of this layer should be 4D.

When you use this layer as the first layer of a model, you need to provide the argument inputShape (a Single Shape, does not include the batch dimension).

T

Numeric type of parameter(e.g. weight, bias). Only support float/double now.

Linear Supertypes
Net, bigdl.nn.keras.BatchNormalization[T], KerasLayer[Tensor[T], Tensor[T], T], Container[Tensor[T], Tensor[T], T], AbstractModule[Tensor[T], Tensor[T], T], InferShape, Serializable, Serializable, AnyRef, Any
Ordering
  1. Alphabetic
  2. By inheritance
Inherited
  1. BatchNormalization
  2. Net
  3. BatchNormalization
  4. KerasLayer
  5. Container
  6. AbstractModule
  7. InferShape
  8. Serializable
  9. Serializable
  10. AnyRef
  11. Any
  1. Hide All
  2. Show all
Learn more about member selection
Visibility
  1. Public
  2. All

Instance Constructors

  1. new BatchNormalization(epsilon: Double = 0.001, momentum: Double = 0.99, betaInit: String = "zero", gammaInit: String = "one", dimOrdering: DataFormat = ..., inputShape: Shape = null)(implicit arg0: ClassTag[T], ev: TensorNumeric[T])

    epsilon

    Small Double > 0. Fuzz parameter. Default is 0.001.

    momentum

    Double. Momentum in the computation of the exponential average of the mean and standard deviation of the data, for feature-wise normalization. Default is 0.99.

    betaInit

    Name of initialization function for shift parameter. Default is 'zero'.

    gammaInit

    Name of initialization function for scale parameter. Default is 'one'.

    dimOrdering

    Format of input data. Either DataFormat.NCHW (dimOrdering='th') or DataFormat.NHWC (dimOrdering='tf'). Default is NCHW. For NCHW, axis along which to normalize is 1. For NHWC, axis is 3.

    inputShape

    A Single Shape, does not include the batch dimension.

Value Members

  1. final def !=(arg0: AnyRef): Boolean

    Definition Classes
    AnyRef
  2. final def !=(arg0: Any): Boolean

    Definition Classes
    Any
  3. final def ##(): Int

    Definition Classes
    AnyRef → Any
  4. final def ==(arg0: AnyRef): Boolean

    Definition Classes
    AnyRef
  5. final def ==(arg0: Any): Boolean

    Definition Classes
    Any
  6. def accGradParameters(input: Tensor[T], gradOutput: Tensor[T]): Unit

    Definition Classes
    KerasLayer → AbstractModule
  7. def apply(name: String): Option[AbstractModule[Activity, Activity, T]]

    Definition Classes
    Container → AbstractModule
  8. final def asInstanceOf[T0]: T0

    Definition Classes
    Any
  9. def backward(input: Tensor[T], gradOutput: Tensor[T]): Tensor[T]

    Definition Classes
    AbstractModule
  10. var backwardTime: Long

    Attributes
    protected
    Definition Classes
    AbstractModule
  11. val betaInit: String

    Name of initialization function for shift parameter.

    Name of initialization function for shift parameter. Default is 'zero'.

    Definition Classes
    BatchNormalization → BatchNormalization
  12. def build(calcInputShape: Shape): Shape

    Definition Classes
    KerasLayer → InferShape
  13. def canEqual(other: Any): Boolean

    Definition Classes
    Container → AbstractModule
  14. final def checkEngineType(): BatchNormalization.this.type

    Definition Classes
    Container → AbstractModule
  15. def clearState(): BatchNormalization.this.type

    Definition Classes
    Container → AbstractModule
  16. final def clone(deepCopy: Boolean): AbstractModule[Tensor[T], Tensor[T], T]

    Definition Classes
    AbstractModule
  17. def clone(): AnyRef

    Attributes
    protected[java.lang]
    Definition Classes
    AnyRef
    Annotations
    @throws( ... )
  18. final def cloneModule(): AbstractModule[Tensor[T], Tensor[T], T]

    Definition Classes
    AbstractModule
  19. def computeOutputShape(inputShape: Shape): Shape

    Definition Classes
    BatchNormalization → KerasLayer → InferShape
  20. val dimOrdering: DataFormat

    Format of input data.

    Format of input data. Either DataFormat.NCHW (dimOrdering='th') or DataFormat.NHWC (dimOrdering='tf'). Default is NCHW. For NCHW, axis along which to normalize is 1. For NHWC, axis is 3.

    Definition Classes
    BatchNormalization → BatchNormalization
  21. def doBuild(inputShape: Shape): AbstractModule[Tensor[T], Tensor[T], T]

    Definition Classes
    BatchNormalization → KerasLayer
  22. val epsilon: Double

    Small Double > 0.

    Small Double > 0. Fuzz parameter. Default is 0.001.

    Definition Classes
    BatchNormalization → BatchNormalization
  23. final def eq(arg0: AnyRef): Boolean

    Definition Classes
    AnyRef
  24. def equals(other: Any): Boolean

    Definition Classes
    Container → AbstractModule → AnyRef → Any
  25. final def evaluate(): BatchNormalization.this.type

    Definition Classes
    Container → AbstractModule
  26. final def evaluate(dataSet: LocalDataSet[MiniBatch[T]], vMethods: Array[_ <: ValidationMethod[T]]): Array[(ValidationResult, ValidationMethod[T])]

    Definition Classes
    AbstractModule
  27. final def evaluate(dataset: RDD[Sample[T]], vMethods: Array[_ <: ValidationMethod[T]], batchSize: Option[Int]): Array[(ValidationResult, ValidationMethod[T])]

    Definition Classes
    AbstractModule
  28. final def evaluateImage(imageFrame: ImageFrame, vMethods: Array[_ <: ValidationMethod[T]], batchSize: Option[Int]): Array[(ValidationResult, ValidationMethod[T])]

    Definition Classes
    AbstractModule
  29. def finalize(): Unit

    Attributes
    protected[java.lang]
    Definition Classes
    AnyRef
    Annotations
    @throws( classOf[java.lang.Throwable] )
  30. def findModules(moduleType: String): ArrayBuffer[AbstractModule[_, _, T]]

    Definition Classes
    Container
  31. final def forward(input: Tensor[T]): Tensor[T]

    Definition Classes
    AbstractModule
  32. var forwardTime: Long

    Attributes
    protected
    Definition Classes
    AbstractModule
  33. def freeze(names: String*): BatchNormalization.this.type

    Definition Classes
    Container → AbstractModule
  34. def from[T](vars: Variable[T]*)(implicit arg0: ClassTag[T], ev: TensorNumeric[T]): Variable[T]

    Build graph: some other modules point to current module

    Build graph: some other modules point to current module

    vars

    upstream variables

    returns

    Variable containing current module

    Definition Classes
    Net
  35. val gammaInit: String

    Name of initialization function for scale parameter.

    Name of initialization function for scale parameter. Default is 'one'.

    Definition Classes
    BatchNormalization → BatchNormalization
  36. final def getClass(): Class[_]

    Definition Classes
    AnyRef → Any
  37. def getExtraParameter(): Array[Tensor[T]]

    Definition Classes
    Container → AbstractModule
  38. final def getInputShape(): Shape

    Definition Classes
    InferShape
  39. final def getName(): String

    Definition Classes
    AbstractModule
  40. final def getNumericType(): TensorDataType

    Definition Classes
    AbstractModule
  41. final def getOutputShape(): Shape

    Definition Classes
    InferShape
  42. def getParametersTable(): Table

    Definition Classes
    Container → AbstractModule
  43. final def getPrintName(): String

    Attributes
    protected
    Definition Classes
    AbstractModule
  44. final def getScaleB(): Double

    Definition Classes
    AbstractModule
  45. final def getScaleW(): Double

    Definition Classes
    AbstractModule
  46. def getTimes(): Array[(AbstractModule[_ <: Activity, _ <: Activity, T], Long, Long)]

    Definition Classes
    Container → AbstractModule
  47. final def getTimesGroupByModuleType(): Array[(String, Long, Long)]

    Definition Classes
    AbstractModule
  48. final def getWeightsBias(): Array[Tensor[T]]

    Definition Classes
    AbstractModule
  49. var gradInput: Tensor[T]

    Definition Classes
    AbstractModule
  50. final def hasName: Boolean

    Definition Classes
    AbstractModule
  51. def hashCode(): Int

    Definition Classes
    Container → AbstractModule → AnyRef → Any
  52. val inputShape: Shape

    A Single Shape, does not include the batch dimension.

    A Single Shape, does not include the batch dimension.

    Definition Classes
    BatchNormalization → BatchNormalization
  53. def inputs(first: (ModuleNode[T], Int), nodesWithIndex: (ModuleNode[T], Int)*): ModuleNode[T]

    Definition Classes
    KerasLayer → AbstractModule
  54. def inputs(nodes: Array[ModuleNode[T]]): ModuleNode[T]

    Definition Classes
    KerasLayer → AbstractModule
  55. def inputs(nodes: ModuleNode[T]*): ModuleNode[T]

    Definition Classes
    KerasLayer → AbstractModule
  56. def isBuilt(): Boolean

    Definition Classes
    KerasLayer → InferShape
  57. def isFrozen[T]()(implicit arg0: ClassTag[T]): Boolean

    Definition Classes
    Net
  58. final def isInstanceOf[T0]: Boolean

    Definition Classes
    Any
  59. def isKerasStyle(): Boolean

    Definition Classes
    KerasLayer → InferShape
  60. final def isTraining(): Boolean

    Definition Classes
    AbstractModule
  61. def labor: AbstractModule[Tensor[T], Tensor[T], T]

    Definition Classes
    KerasLayer
  62. def labor_=(value: AbstractModule[Tensor[T], Tensor[T], T]): Unit

    Definition Classes
    KerasLayer
  63. var line: String

    Attributes
    protected
    Definition Classes
    AbstractModule
  64. final def loadModelWeights(srcModel: Module[Float], matchAll: Boolean): BatchNormalization.this.type

    Definition Classes
    AbstractModule
  65. final def loadWeights(weightPath: String, matchAll: Boolean): BatchNormalization.this.type

    Definition Classes
    AbstractModule
  66. val modules: ArrayBuffer[AbstractModule[Activity, Activity, T]]

    Definition Classes
    Container
  67. val momentum: Double

    Double.

    Double. Momentum in the computation of the exponential average of the mean and standard deviation of the data, for feature-wise normalization. Default is 0.99.

    Definition Classes
    BatchNormalization → BatchNormalization
  68. final def ne(arg0: AnyRef): Boolean

    Definition Classes
    AnyRef
  69. final def notify(): Unit

    Definition Classes
    AnyRef
  70. final def notifyAll(): Unit

    Definition Classes
    AnyRef
  71. var output: Tensor[T]

    Definition Classes
    AbstractModule
  72. def parameters(): (Array[Tensor[T]], Array[Tensor[T]])

    Definition Classes
    Container → AbstractModule
  73. final def predict(dataset: RDD[Sample[T]], batchSize: Int, shareBuffer: Boolean): RDD[Activity]

    Definition Classes
    AbstractModule
  74. final def predictClass(dataset: RDD[Sample[T]], batchSize: Int): RDD[Int]

    Definition Classes
    AbstractModule
  75. final def predictImage(imageFrame: ImageFrame, outputLayer: String, shareBuffer: Boolean, batchPerPartition: Int, predictKey: String, featurePaddingParam: Option[PaddingParam[T]]): ImageFrame

    Definition Classes
    AbstractModule
  76. def processInputs(first: (ModuleNode[T], Int), nodesWithIndex: (ModuleNode[T], Int)*): ModuleNode[T]

    Attributes
    protected
    Definition Classes
    AbstractModule
  77. def processInputs(nodes: Seq[ModuleNode[T]]): ModuleNode[T]

    Attributes
    protected
    Definition Classes
    AbstractModule
  78. final def quantize(): Module[T]

    Definition Classes
    AbstractModule
  79. def release(): Unit

    Definition Classes
    Container → AbstractModule
  80. def reset(): Unit

    Definition Classes
    Container → AbstractModule
  81. def resetTimes(): Unit

    Definition Classes
    Container → AbstractModule
  82. final def saveCaffe(prototxtPath: String, modelPath: String, useV2: Boolean, overwrite: Boolean): BatchNormalization.this.type

    Definition Classes
    AbstractModule
  83. final def saveDefinition(path: String, overWrite: Boolean): BatchNormalization.this.type

    Definition Classes
    AbstractModule
  84. final def saveModule(path: String, weightPath: String, overWrite: Boolean): BatchNormalization.this.type

    Definition Classes
    AbstractModule
  85. final def saveTF(inputs: Seq[(String, Seq[Int])], path: String, byteOrder: ByteOrder, dataFormat: TensorflowDataFormat): BatchNormalization.this.type

    Definition Classes
    AbstractModule
  86. final def saveTorch(path: String, overWrite: Boolean): BatchNormalization.this.type

    Definition Classes
    AbstractModule
  87. final def saveWeights(path: String, overWrite: Boolean): Unit

    Definition Classes
    AbstractModule
  88. var scaleB: Double

    Attributes
    protected
    Definition Classes
    AbstractModule
  89. var scaleW: Double

    Attributes
    protected
    Definition Classes
    AbstractModule
  90. final def setExtraParameter(extraParam: Array[Tensor[T]]): BatchNormalization.this.type

    Definition Classes
    AbstractModule
  91. final def setLine(line: String): BatchNormalization.this.type

    Definition Classes
    AbstractModule
  92. final def setName(name: String): BatchNormalization.this.type

    Definition Classes
    AbstractModule
  93. def setScaleB(b: Double): BatchNormalization.this.type

    Definition Classes
    Container → AbstractModule
  94. def setScaleW(w: Double): BatchNormalization.this.type

    Definition Classes
    Container → AbstractModule
  95. final def setWeightsBias(newWeights: Array[Tensor[T]]): BatchNormalization.this.type

    Definition Classes
    AbstractModule
  96. final def synchronized[T0](arg0: ⇒ T0): T0

    Definition Classes
    AnyRef
  97. def toGraph(startNodes: ModuleNode[T]*): Graph[T]

    Definition Classes
    AbstractModule
  98. def toString(): String

    Definition Classes
    AbstractModule → AnyRef → Any
  99. var train: Boolean

    Attributes
    protected
    Definition Classes
    AbstractModule
  100. final def training(): BatchNormalization.this.type

    Definition Classes
    Container → AbstractModule
  101. def unFreeze(names: String*): BatchNormalization.this.type

    Definition Classes
    Container → AbstractModule
  102. def updateGradInput(input: Tensor[T], gradOutput: Tensor[T]): Tensor[T]

    Definition Classes
    KerasLayer → AbstractModule
  103. def updateOutput(input: Tensor[T]): Tensor[T]

    Definition Classes
    KerasLayer → AbstractModule
  104. final def wait(): Unit

    Definition Classes
    AnyRef
    Annotations
    @throws( ... )
  105. final def wait(arg0: Long, arg1: Int): Unit

    Definition Classes
    AnyRef
    Annotations
    @throws( ... )
  106. final def wait(arg0: Long): Unit

    Definition Classes
    AnyRef
    Annotations
    @throws( ... )
  107. def zeroGradParameters(): Unit

    Definition Classes
    AbstractModule

Deprecated Value Members

  1. final def save(path: String, overWrite: Boolean): BatchNormalization.this.type

    Definition Classes
    AbstractModule
    Annotations
    @deprecated
    Deprecated

    (Since version 0.3.0) please use recommended saveModule(path, overWrite)

Inherited from Net

Inherited from bigdl.nn.keras.BatchNormalization[T]

Inherited from KerasLayer[Tensor[T], Tensor[T], T]

Inherited from Container[Tensor[T], Tensor[T], T]

Inherited from AbstractModule[Tensor[T], Tensor[T], T]

Inherited from InferShape

Inherited from Serializable

Inherited from Serializable

Inherited from AnyRef

Inherited from Any

Ungrouped